Skip to content

Commit 09bdef0

Browse files
C-Achardvidalmaxime
authored andcommitted
Fixes + reqs update
- Fixed model instantiation - Fix window overlap argument type error - Updated reqs.txt to have einops (MONAI optional dep.)
1 parent 70a936a commit 09bdef0

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

napari_cellseg3d/models/model_SwinUNetR.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44

55
def get_weights_file():
6-
return ""
6+
return "Swin64_best_metric.pth"
77

88

99
def get_net(img_size, use_checkpoint=True):

requirements.txt

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,13 +13,10 @@ napari[all]>=0.4.14
1313
QtPy
1414
opencv-python>=4.5.5
1515
dask-image>=0.6.0
16-
scikit-image>=0.19.2
1716
matplotlib>=3.4.1
1817
tifffile>=2022.2.9
1918
imageio-ffmpeg>=0.4.5
2019
torch>=1.11
21-
monai>=0.9.0
22-
nibabel
20+
monai[nibabel,scikit-image,itk,einops]>=0.9.0
2321
pillow
24-
itk>=5.2.0
2522
vispy>=0.9.6

0 commit comments

Comments
 (0)