add requirements.txt

.gitignore

@@ -5,7 +5,7 @@ weights/
 x264/
 *.mp3
 *.mp4
-*.txt
+filelist.txt
 *.png
 complex_infer.sh
 __pycache__/

infer.sh

@@ -1,7 +1,7 @@
 rm -r "./results"
 mkdir "./results"
 
-rlaunch --gpu=1 --cpu=4 --memory=25600 -- python3 -m torch.distributed.launch \
+python3 -m torch.distributed.launch \
 --nproc_per_node=1 train.py --mode=test \
 --world_size=1 --dataloaders=2 \
 --test_input_poses_images=./test_data/ \

requirements.txt

@@ -0,0 +1,8 @@
+numpy>=1.16
+tqdm>=4.35.0
+torch>=1.3.0
+opencv-python>=4.5.2
+scikit-image>=0.14.0
+torchvision>=0.2.1
+pillow>=7.2.0
+lpips>=0.1.3