Spaces:
Runtime error
Runtime error
added some debug lines
Browse files
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 |
-
|
|
|
|
| 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)
|