Ivan Felipe Rodriguez commited on
Commit
abe0334
·
1 Parent(s): 6eca86c

removing pdb

Browse files
Files changed (1) hide show
  1. 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
- import pdb;pdb.set_trace()
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