ianpan commited on
Commit
9fef83a
·
verified ·
1 Parent(s): 9d7de53

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
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)