Update README.md
Browse files
README.md
CHANGED
|
@@ -86,6 +86,8 @@ coords = coords[0].cpu().numpy()
|
|
| 86 |
x, y, w, h = coords
|
| 87 |
# coords already rescaled with img_shape
|
| 88 |
cropped_img = img[y: y + h, x: x + w]
|
|
|
|
|
|
|
| 89 |
|
| 90 |
model = AutoModel.from_pretrained("ianpan/bone-age", trust_remote_code=True)
|
| 91 |
model = model.eval().to(device)
|
|
|
|
| 86 |
x, y, w, h = coords
|
| 87 |
# coords already rescaled with img_shape
|
| 88 |
cropped_img = img[y: y + h, x: x + w]
|
| 89 |
+
ref = cv2.imread("ref_img.png", 0) # download ref_img.png from this repo
|
| 90 |
+
cropped_img = match_histograms(cropped_img, ref)
|
| 91 |
|
| 92 |
model = AutoModel.from_pretrained("ianpan/bone-age", trust_remote_code=True)
|
| 93 |
model = model.eval().to(device)
|