Ahsen Khaliq commited on
Commit
b1e3471
1 Parent(s): 8969b51

Update tools/demo_track.py

Browse files
Files changed (1) hide show
  1. tools/demo_track.py +1 -5
tools/demo_track.py CHANGED
@@ -207,11 +207,7 @@ def image_demo(predictor, vis_folder, path, current_time, save_result):
207
 
208
  #result_image = predictor.visual(outputs[0], img_info, predictor.confthre)
209
  if save_result:
210
- save_folder = os.path.join(
211
- vis_folder, time.strftime("%Y_%m_%d_%H_%M_%S", current_time)
212
- )
213
- os.makedirs(save_folder, exist_ok=True)
214
- save_file_name = os.path.join(save_folder, os.path.basename(image_name))
215
  cv2.imwrite(save_file_name, online_im)
216
  ch = cv2.waitKey(0)
217
  frame_id += 1
 
207
 
208
  #result_image = predictor.visual(outputs[0], img_info, predictor.confthre)
209
  if save_result:
210
+ save_file_name = "out.jpg"
 
 
 
 
211
  cv2.imwrite(save_file_name, online_im)
212
  ch = cv2.waitKey(0)
213
  frame_id += 1