Dockerfile now uses the same train_log folder as the scripts

This commit is contained in:
ko1N
2021-01-17 16:07:54 +01:00
parent 0f980597a1
commit f455a05736
2 changed files with 2 additions and 2 deletions

View File

@@ -81,7 +81,7 @@ ffmpeg -r 10 -f image2 -i output/img%d.png -s 448x256 -vf "split[s0][s1];[s0]pal
``` ```
### Run in docker ### Run in docker
Place the pre-trained models in the `./docker/pretrained_models directory` Place the pre-trained models in `train_log/\*.pkl` (as above)
Building the container: Building the container:
``` ```

View File

@@ -15,7 +15,7 @@ ADD docker/inference_video /usr/local/bin/inference_video
RUN chmod +x /usr/local/bin/inference_video RUN chmod +x /usr/local/bin/inference_video
# add pre-trained models # add pre-trained models
COPY docker/pretrained_models /rife/train_log COPY train_log /rife/train_log
WORKDIR /host WORKDIR /host
ENTRYPOINT ["/bin/bash"] ENTRYPOINT ["/bin/bash"]