Commit c391aef9 authored by Rafael Dätwyler's avatar Rafael Dätwyler
Browse files

start indexing of frames at 0 instead of 1

parent 850d0f1b
......@@ -40,7 +40,7 @@ device = torch.device("cuda" if cuda else "cpu")
# Parse arguments
n_videos = 9
n_frames = 13000-110
n_frames = 13000-111
n_latents = 64
# Load model
......@@ -64,7 +64,7 @@ with torch.no_grad():
# Extract video number
tmp = os.path.splitext(imgfilename)[0].split("_")
video_n = int(tmp[0])
frame_n = int(tmp[2]) - 110
frame_n = int(tmp[2]) - 111
img = transforms(rgba_loader(join("train/nolabel",imgfilename)))
# Run throug model encoder
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment