Ivan Felipe Rodriguez commited on
Commit
cf97cd7
·
1 Parent(s): 7abf939

removing pdb

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -72,7 +72,7 @@ classes= [ 'Beach',
72
  'Oceanic whitetip shark',
73
  'Blacktip shark',
74
  'Tiger shark',
75
- 'Bull shark']
76
 
77
 
78
 
@@ -101,7 +101,7 @@ register_all_modules()
101
  # build the model from a config file and a checkpoint file
102
  model = init_detector(config_file, checkpoint_file, device='cuda:0') # or device='cuda:0'
103
  model.dataset_meta['classes'] = classes
104
- model.dataset_meta['thing_classes'] = classes*2 [:len(model.dataset_meta['thing_classes'])]
105
 
106
  model.dataset_meta['palette'] = model.dataset_meta['palette']*3
107
 
 
72
  'Oceanic whitetip shark',
73
  'Blacktip shark',
74
  'Tiger shark',
75
+ 'Bull shark']*2
76
 
77
 
78
 
 
101
  # build the model from a config file and a checkpoint file
102
  model = init_detector(config_file, checkpoint_file, device='cuda:0') # or device='cuda:0'
103
  model.dataset_meta['classes'] = classes
104
+ model.dataset_meta['thing_classes'] = classes [:len(model.dataset_meta['thing_classes'])]
105
 
106
  model.dataset_meta['palette'] = model.dataset_meta['palette']*3
107