Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -44,7 +44,7 @@ model_name_to_id = {model["name"] : id_ for id_, model in enumerate(models)}
|
|
| 44 |
for model in models:
|
| 45 |
|
| 46 |
model["cfg"] = get_cfg()
|
| 47 |
-
model["cfg"].merge_from_file("./configs/detectron2/
|
| 48 |
model["cfg"].MODEL.ROI_HEADS.NUM_CLASSES = len(model["classes"])
|
| 49 |
model["cfg"].MODEL.WEIGHTS = model["model_path"]
|
| 50 |
|
|
|
|
| 44 |
for model in models:
|
| 45 |
|
| 46 |
model["cfg"] = get_cfg()
|
| 47 |
+
model["cfg"].merge_from_file("./configs/detectron2/mask_rcnn_X_101_32x8d_FPN_3x.yaml")
|
| 48 |
model["cfg"].MODEL.ROI_HEADS.NUM_CLASSES = len(model["classes"])
|
| 49 |
model["cfg"].MODEL.WEIGHTS = model["model_path"]
|
| 50 |
|