iicd/cotracker
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix ignored input video argument (#57)

Browse Source
This commit is contained in:
Patrick Pfreundschuh 2024-01-07 16:14:28 +01:00 committed by GitHub
parent 941c24fd40
commit 9921cf0895
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
online_demo.py
View File

@ -4,6 +4,7 @@
# This source code is licensed under the license found in the # This source code is licensed under the license found in the
# LICENSE file in the root directory of this source tree. # LICENSE file in the root directory of this source tree.
import os
import torch import torch
import argparse import argparse
import imageio.v3 as iio import imageio.v3 as iio
@ -44,6 +45,9 @@ if __name__ == "__main__":
args = parser.parse_args() args = parser.parse_args()
if not os.path.isfile(args.video_path):
raise ValueError("Video file does not exist")
if args.checkpoint is not None: if args.checkpoint is not None:
model = CoTrackerOnlinePredictor(checkpoint=args.checkpoint) model = CoTrackerOnlinePredictor(checkpoint=args.checkpoint)
else: else:
@ -69,7 +73,7 @@ if __name__ == "__main__":
is_first_step = True is_first_step = True
for i, frame in enumerate( for i, frame in enumerate(
iio.imiter( iio.imiter(
"./assets/apple.mp4", args.video_path,
plugin="FFMPEG", plugin="FFMPEG",
) )
): ):