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