Added ability to run both inference scripts from docker

README.md

@@ -90,7 +90,10 @@ docker build -t rife -f docker/Dockerfile .
 
 Running the container:
 ```
-docker run --rm -it -v $PWD:/host rife:latest --exp=1 --video=untitled.mp4 --output=untitled_rife.mp4
+docker run --rm -it -v $PWD:/host rife:latest inference_video --exp=1 --video=untitled.mp4 --output=untitled_rife.mp4
+```
+```
+docker run --rm -it -v $PWD:/host rife:latest inference_img --img img0.png img1.png --exp=4
 ```
 
 ## Evaluation

docker/Dockerfile

@@ -9,13 +9,15 @@ WORKDIR /rife
 COPY . .
 RUN pip3 install -r requirements.txt
 
-ADD docker/rife.sh /usr/local/bin/rife
-RUN chmod +x /usr/local/bin/rife
+ADD docker/inference_img /usr/local/bin/inference_img
+RUN chmod +x /usr/local/bin/inference_img
+ADD docker/inference_video /usr/local/bin/inference_video
+RUN chmod +x /usr/local/bin/inference_video
 
 # add pre-trained models
 COPY docker/pretrained_models /rife/train_log
 
 WORKDIR /host
-ENTRYPOINT ["rife"]
+ENTRYPOINT ["/bin/bash"]
 
 ENV NVIDIA_DRIVER_CAPABILITIES all

docker/inference_img

@@ -0,0 +1,2 @@
+#!/bin/sh
+python3 /rife/inference_img.py $@