piperod91 commited on
Commit
9bc9c4c
1 Parent(s): 2f37c23

fixing label

Browse files
Files changed (1) hide show
  1. inference.py +3 -3
inference.py CHANGED
@@ -58,8 +58,8 @@ classes = ['Beach',
58
  'Blue shark',
59
  'Great white shark',
60
  'Shark',
61
- 'Nurse shark',
62
- 'Silky shark',
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 detector
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