Skip to content

Commit

Permalink
bug fix: skip downloading videos for existing folders in database/raw (
Browse files Browse the repository at this point in the history
  • Loading branch information
DenisKochetov authored Sep 13, 2023
1 parent 7d72404 commit 24ed2d2
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions scripts/run_preprocess.py
Original file line number Diff line number Diff line change
Expand Up @@ -114,8 +114,11 @@ def run_extract_priors(seqname, outdir, obj_class_cam):
print("using gpus: ", gpulist)
os.makedirs("tmp", exist_ok=True)

# download the videos
download_seq(vidname)
# check if the directory with the video already exists
viddir_path = os.path.join("database", "raw", vidname)
if not os.path.exists(viddir_path):
# download the videos only if the directory does not exist
download_seq(vidname)

# set up parallel extraction
frame_args = []
Expand Down

0 comments on commit 24ed2d2

Please sign in to comment.