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

removing pdb

Browse files
Files changed (1) hide show
  1. inference.py +2 -2
inference.py CHANGED
@@ -100,8 +100,8 @@ register_all_modules()
100
 
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*2
104
- model.dataset_meta['thing_classes'] = classes*2
105
 
106
  model.dataset_meta['palette'] = model.dataset_meta['palette']*3
107
 
 
100
 
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