Model Description in apps/mmars updated with NGC description for Deepgrow
@@ -184,4 +184,26 @@ MODEL_DESC = (
RemoteMMARKeys.HASH_VAL: None,
RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"),
},
+ {
+ RemoteMMARKeys.ID: "clara_pt_deepgrow_3d_annotation_1",
+ RemoteMMARKeys.NAME: "clara_pt_deepgrow_3d_annotation",
+ RemoteMMARKeys.URL: "https://api.ngc.nvidia.com/v2/models/nvidia/med/clara_pt_deepgrow_3d_annotation/versions/1/zip",
+ RemoteMMARKeys.DOC: "https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_deepgrow_3d_annotation",
+ RemoteMMARKeys.FILE_TYPE: "zip",
+ RemoteMMARKeys.HASH_TYPE: "md5",
+ RemoteMMARKeys.HASH_VAL: None,
+ RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"),
+ RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"),
+ },
+ {
+ RemoteMMARKeys.ID: "clara_pt_deepgrow_2d_annotation_1",
+ RemoteMMARKeys.NAME: "clara_pt_deepgrow_2d_annotation",
+ RemoteMMARKeys.URL: "https://api.ngc.nvidia.com/v2/models/nvidia/med/clara_pt_deepgrow_2d_annotation/versions/1/zip",
+ RemoteMMARKeys.DOC: "https://ngc.nvidia.com/catalog/models/nvidia:med:clara_pt_deepgrow_2d_annotation",
+ RemoteMMARKeys.FILE_TYPE: "zip",
+ RemoteMMARKeys.HASH_TYPE: "md5",
+ RemoteMMARKeys.HASH_VAL: None,
+ RemoteMMARKeys.MODEL_FILE: os.path.join("models", "model.pt"),
+ RemoteMMARKeys.CONFIG_FILE: os.path.join("config", "config_train.json"),
+ },
)
