Spaces:
Runtime error
Runtime error
Ivan Felipe Rodriguez
commited on
Commit
·
abe0334
1
Parent(s):
6eca86c
removing pdb
Browse files- inference.py +2 -3
inference.py
CHANGED
@@ -103,6 +103,7 @@ model = init_detector(config_file, checkpoint_file, device='cuda:0') # or devic
|
|
103 |
model.dataset_meta['classes'] = classes
|
104 |
model.dataset_meta['thing_classes'] = classes
|
105 |
model.dataset_meta['stuff_classes'] = []
|
|
|
106 |
|
107 |
print(model.cfg.visualizer)
|
108 |
# init visualizer(run the block only once in jupyter notebook)
|
@@ -111,13 +112,11 @@ print(dir(visualizer))
|
|
111 |
# the dataset_meta is loaded from the checkpoint and
|
112 |
# then pass to the model in init_detector
|
113 |
visualizer.dataset_meta = model.dataset_meta
|
114 |
-
|
115 |
|
116 |
def inference_frame(image):
|
117 |
-
#import ipdb; ipdb.set_trace()
|
118 |
result = inference_detector(model, image)
|
119 |
# show the results
|
120 |
-
#import ipdb; ipdb.set_trace()
|
121 |
frames = []
|
122 |
cnt=0
|
123 |
|
|
|
103 |
model.dataset_meta['classes'] = classes
|
104 |
model.dataset_meta['thing_classes'] = classes
|
105 |
model.dataset_meta['stuff_classes'] = []
|
106 |
+
model.dataset_meta['palette'] = model.dataset_meta['palette']*3
|
107 |
|
108 |
print(model.cfg.visualizer)
|
109 |
# init visualizer(run the block only once in jupyter notebook)
|
|
|
112 |
# the dataset_meta is loaded from the checkpoint and
|
113 |
# then pass to the model in init_detector
|
114 |
visualizer.dataset_meta = model.dataset_meta
|
115 |
+
|
116 |
|
117 |
def inference_frame(image):
|
|
|
118 |
result = inference_detector(model, image)
|
119 |
# show the results
|
|
|
120 |
frames = []
|
121 |
cnt=0
|
122 |
|