oskarastrom commited on
Commit
b02b8a0
1 Parent(s): 5657a6c

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +3 -3
inference.py CHANGED
@@ -80,11 +80,11 @@ def do_full_inference(dataloader, image_meter_width, image_meter_height, gp=None
80
 
81
  outputs = do_suppression(inference, gp=gp)
82
 
83
- do_confidence_boost(inference, outputs, gp=gp)
84
 
85
- new_outputs = do_suppression(inference, gp=gp)
86
 
87
- all_preds, real_width, real_height = format_predictions(image_shapes, new_outputs, width, height, gp=gp)
88
 
89
  results = do_tracking(all_preds, image_meter_width, image_meter_height, gp=gp)
90
 
 
80
 
81
  outputs = do_suppression(inference, gp=gp)
82
 
83
+ #do_confidence_boost(inference, outputs, gp=gp)
84
 
85
+ #new_outputs = do_suppression(inference, gp=gp)
86
 
87
+ all_preds, real_width, real_height = format_predictions(image_shapes, outputs, width, height, gp=gp)
88
 
89
  results = do_tracking(all_preds, image_meter_width, image_meter_height, gp=gp)
90