aharley commited on
Commit
a8f5bea
·
1 Parent(s): a7bc06c

added some debug lines

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -527,7 +527,8 @@ def update_vis(rate, show_bkg, cmap, video_preview, query_frame, video_fps, trac
527
  print('rate', rate)
528
  print('cmap', cmap)
529
  print('video_preview', video_preview.shape)
530
- T, H, W,_ = video_preview.shape
 
531
  tracks_ = tracks.reshape(H,W,T,2)[::rate,::rate].reshape(-1,T,2)
532
  visibs_ = visibs.reshape(H,W,T)[::rate,::rate].reshape(-1,T)
533
  return paint_video(video_preview, query_frame, video_fps, tracks_, visibs_, rate=rate, show_bkg=show_bkg, cmap=cmap)
@@ -794,6 +795,8 @@ def track(
794
  # visibs = (visibs * confs) > 0.1 # N,T
795
  visibs = (confs) > 0.1 # N,T
796
  # visibs = (confs) > 0.1 # N,T
 
 
797
 
798
 
799
  # sc = (np.array([video_preview.shape[2], video_preview.shape[1]]) / np.array([VIDEO_INPUT_RESO[1], VIDEO_INPUT_RESO[0]])).reshape(1,1,2)
 
527
  print('rate', rate)
528
  print('cmap', cmap)
529
  print('video_preview', video_preview.shape)
530
+ print('tracks', tracks.shape)
531
+ T, H, W, _ = video_preview.shape
532
  tracks_ = tracks.reshape(H,W,T,2)[::rate,::rate].reshape(-1,T,2)
533
  visibs_ = visibs.reshape(H,W,T)[::rate,::rate].reshape(-1,T)
534
  return paint_video(video_preview, query_frame, video_fps, tracks_, visibs_, rate=rate, show_bkg=show_bkg, cmap=cmap)
 
795
  # visibs = (visibs * confs) > 0.1 # N,T
796
  visibs = (confs) > 0.1 # N,T
797
  # visibs = (confs) > 0.1 # N,T
798
+
799
+ print('tracks', tracks.shape)
800
 
801
 
802
  # sc = (np.array([video_preview.shape[2], video_preview.shape[1]]) / np.array([VIDEO_INPUT_RESO[1], VIDEO_INPUT_RESO[0]])).reshape(1,1,2)