Update inference.py

inference.py

@@ -249,6 +249,11 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, c
     if (gp): gp(0, "Confidence Boost...")
     # keep predictions to feed them ordered into the Tracker
     # TODO: how to deal with large files?
+
+
+    boost_cutoff = 0.01
+    boost_range = math.floor(math.sqrt(1/conf_decay * math.log(conf_power / boost_cutoff)))
+    
     outputs = []
     with tqdm(total=len(inference), desc="Running confidence boost", ncols=0, disable=not verbose) as pbar:
         for batch_i in range(len(inference)):
@@ -266,9 +271,9 @@ def do_confidence_boost(inference, safe_preds, gp=None, batch_size=BATCH_SIZE, c
                 has_next_batch = batch_i+1 < len(inference)
                 has_prev_batch = batch_i-1 >= 0
 
-                frames = [None, None]
                 
-                for dt in [-1, 1]:
+                for dt in range(-boost_range, boost_range+1):
+                    if dt == 0: continue
                     idx = i+dt 
                     temp_frame = None
                     if idx >= 0 and idx < len(infer):