Skip to content

Commit b041de1

Browse files
committed
Model Description in apps/mmars updated with NGC description for Deepgrow
Signed-off-by: vnath <vnath@nvidia.com>
1 parent 0662ed2 commit b041de1

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

monai/apps/mmars/model_desc.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,4 +129,26 @@ class RemoteMMARKeys:
129129
RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"),
130130
RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"),
131131
},
132+
{
133+
RemoteMMARKeys.ID: "clara_pt_deepgrow_3d_annotation_1",
134+
RemoteMMARKeys.NAME: "clara_pt_deepgrow_3d_annotation",
135+
RemoteMMARKeys.URL: "https://api.ngc.nvidia.com/v2/models/nvidia/med/clara_pt_deepgrow_3d_annotation/versions/1/zip",
136+
RemoteMMARKeys.DOC: "https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_deepgrow_3d_annotation",
137+
RemoteMMARKeys.FILE_TYPE: "zip",
138+
RemoteMMARKeys.HASH_TYPE: "md5",
139+
RemoteMMARKeys.HASH_VAL: None,
140+
RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"),
141+
RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"),
142+
},
143+
{
144+
RemoteMMARKeys.ID: "clara_pt_deepgrow_2d_annotation_1",
145+
RemoteMMARKeys.NAME: "clara_pt_deepgrow_2d_annotation",
146+
RemoteMMARKeys.URL: "https://api.ngc.nvidia.com/v2/models/nvidia/med/clara_pt_deepgrow_2d_annotation/versions/1/zip",
147+
RemoteMMARKeys.DOC: "https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_deepgrow_2d_annotation",
148+
RemoteMMARKeys.FILE_TYPE: "zip",
149+
RemoteMMARKeys.HASH_TYPE: "md5",
150+
RemoteMMARKeys.HASH_VAL: None,
151+
RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"),
152+
RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"),
153+
},
132154
)

0 commit comments

Comments
 (0)