Spaces:
Runtime error
Runtime error
Ivan Felipe Rodriguez
commited on
Commit
·
7abf939
1
Parent(s):
d38714b
removing pdb
Browse files- 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
|
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 |
|