Skip to content

Commit f6a81bf

Browse files
committed
requirements updated
1 parent 6eaf4fa commit f6a81bf

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

README.md

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,11 @@
44
## REQUERIMIENTS
55

66
```bash
7-
conda install pytorch=1.7.1 torchvision=0.8.2 cudatoolkit=11 -c pytorch
7+
conda create --name VideoInp python=3.7.10
8+
conda activate VideoInp
9+
#adding channel nvidia to install exact version of cudatoolkit
10+
conda config --add channels nvidia
11+
conda install pytorch=1.7.1 torchvision=0.8.2 cudatoolkit=11.2.2 -c pytorch
812
pip install -r requirements.txt
913
```
1014

@@ -35,7 +39,7 @@ in video_frames/tennis), runs the following script
3539

3640
```bash
3741
cd ingestion
38-
python create_dataset.py --in_root_dir ../video_frames --out_dir ../dataset --masking_mode same_template --template_mask ../video_frames/tennis/mask.png --compute_RAFT_flow --apply_mask_before --H 256 --W 480 --nLevels 2
42+
`_python create_dataset.py --in_root_dir ../video_frames --out_dir ../dataset --masking_mode same_template --template_mask ../video_frames/tennis/mask.png --apply_mask_before --H 256 --W 480 --nLevels 2_`
3943
```
4044

4145
## Running the overfitting training over data in `./datasets`

0 commit comments

Comments
 (0)