Image Segmentation
medical
biology
Jose commited on
Commit
cc3cca1
·
1 Parent(s): 3ffdf0a

bugfix and python ver bumpup

Browse files
setup.py CHANGED
@@ -23,11 +23,11 @@ setup(
23
  ]
24
  },
25
  install_requires=[
26
- "numpy == 1.*",
27
  "pandas == 2.*",
28
  "tqdm == 4.*",
29
- "Pillow == 9.*",
30
  "click==8.*",
31
  ],
32
- python_requires=">=3.10, <3.11",
33
  )
 
23
  ]
24
  },
25
  install_requires=[
26
+ "numpy == 2.*",
27
  "pandas == 2.*",
28
  "tqdm == 4.*",
29
+ "Pillow == 11.*",
30
  "click==8.*",
31
  ],
32
+ python_requires=">=3.10, <3.13",
33
  )
vascx_models/__init__.py ADDED
File without changes
vascx_models/inference.py CHANGED
@@ -244,7 +244,7 @@ def run_fovea_detection(
244
  heatmap = ensemble_fovea.forward(im)
245
  keypoints = extract_keypoints_from_heatmaps(heatmap)
246
 
247
- kp_fovea = torch.mean(keypoints, dim=0) # average over models
248
 
249
  items = {
250
  "id": batch["id"],
 
244
  heatmap = ensemble_fovea.forward(im)
245
  keypoints = extract_keypoints_from_heatmaps(heatmap)
246
 
247
+ kp_fovea = torch.mean(keypoints, dim=1) # average over models
248
 
249
  items = {
250
  "id": batch["id"],