File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change 7
7
# b0_transversal_5_5 is the name of entrypoint
8
8
def b0_transversal_5_5 (** kwargs ):
9
9
""" # This docstring shows up in hub.help()
10
- Initial model trained on 138 transversal images via transfer learning on the base model chfc-cmi/cmr-seg-tl : cmr_seg_base
10
+ Initial model trained on 138 transversal images via transfer learning on the base model chfc-cmi/cmr-seg-tl : cmr_seg_base with fastai v2
11
11
"""
12
12
# Call the model, load pretrained weights
13
- url = "https://github.com/chfc-cmi/transversal-cmr-seg/releases/download/0.2.0/b0_transversal_5_5 .pth"
14
- dst = torch .hub .get_dir () + '/chfc-cmi_transversal-cmr-seg_master/' + 'b0_transversal_5_5 .pth'
13
+ url = "https://github.com/chfc-cmi/transversal-cmr-seg/releases/download/0.2.0/b0_transversal_5_5_v2 .pth"
14
+ dst = torch .hub .get_dir () + '/chfc-cmi_transversal-cmr-seg_master/' + 'b0_transversal_5_5_v2 .pth'
15
15
if not os .path .isfile (dst ):
16
16
torch .hub .download_url_to_file (url ,dst ,progress = True )
17
17
model = fastai .learner .load_learner (path = tempfile .gettempdir (),file = dst ,tfm_y = False )
You can’t perform that action at this time.
0 commit comments