Update detect_frames.py

scripts/detect_frames.py

@@ -77,6 +77,7 @@ def detect(in_seq_dir, out_seq_dir, model, device, verbose):
             else:
                 inference_list.append(tensor)
             pbar.update(1)
+    pbar.close()
 
     print('converted tensor')
     json_obj = {
@@ -85,10 +86,13 @@ def detect(in_seq_dir, out_seq_dir, model, device, verbose):
         'width': width,
         'height': height
     }
+    print('created object')
     json_string = json.dumps(json_obj)
+    print('created string')
 
     with open(os.path.join(out_seq_dir, 'pred.json'), 'w') as f:
         f.write(json_string)
+    print('saved string')
 
 def argument_parser():
     parser = argparse.ArgumentParser()