Commit d87b0574 authored by nstorni's avatar nstorni
Browse files

Merge branch 'development' of gitlab.ethz.ch:deep-learning-rodent/world-models into development

parents 1d029b9c c391aef9
......@@ -180,7 +180,8 @@ Load one video to $SCRATCH/data/$DATASET_NAME/video using winSCP or similar.
E.g copy from local (ON YOUR LOCAL MACHINE) directory video.avi to the target directory in the cloud.:
```bash
scp *.mp4 nstorni@login.leonhard.ethz.ch:$SCRATCH/data/mini_mice_dataset/video
USERNAME="your-username"
scp *.mp4 $USERNAME@login.leonhard.ethz.ch:$SCRATCH/data/$DATASET_NAME/video
```
Generate interframe differences:
......
......@@ -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