Spaces:
Runtime error
Runtime error
fixing label
Browse files- inference.py +3 -3
inference.py
CHANGED
@@ -58,8 +58,8 @@ classes = ['Beach',
|
|
58 |
'Blue shark',
|
59 |
'Great white shark',
|
60 |
'Shark',
|
61 |
-
'
|
62 |
-
'
|
63 |
'Leopard shark',
|
64 |
'Shortfin mako shark',
|
65 |
'Hammerhead shark',
|
@@ -138,7 +138,7 @@ if not os.path.exists('model'):
|
|
138 |
FILENAME = "mask2former"
|
139 |
snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')
|
140 |
|
141 |
-
# Choose to use a config and initialize the
|
142 |
config_file ='model/mask2former_swin-t-p4-w7-224_8xb2-lsj-50e_coco-panoptic/mask2former_swin-t-p4-w7-224_8xb2-lsj-50e_coco-panoptic.py'
|
143 |
#'/content/mmdetection/configs/panoptic_fpn/panoptic-fpn_r50_fpn_ms-3x_coco.py'
|
144 |
# Setup a checkpoint file to load
|
|
|
58 |
'Blue shark',
|
59 |
'Great white shark',
|
60 |
'Shark',
|
61 |
+
'N shark',
|
62 |
+
'S shark',
|
63 |
'Leopard shark',
|
64 |
'Shortfin mako shark',
|
65 |
'Hammerhead shark',
|
|
|
138 |
FILENAME = "mask2former"
|
139 |
snapshot_download(repo_id=REPO_ID, token= os.environ.get('SHARK_MODEL'),local_dir='model/')
|
140 |
|
141 |
+
# Choose to use a config and initialize the detectorN
|
142 |
config_file ='model/mask2former_swin-t-p4-w7-224_8xb2-lsj-50e_coco-panoptic/mask2former_swin-t-p4-w7-224_8xb2-lsj-50e_coco-panoptic.py'
|
143 |
#'/content/mmdetection/configs/panoptic_fpn/panoptic-fpn_r50_fpn_ms-3x_coco.py'
|
144 |
# Setup a checkpoint file to load
|