From 75f691574d0cd08f8b2cbe8e967c3eb71392627b Mon Sep 17 00:00:00 2001 From: Askat Kuzdeuov Date: Fri, 10 Sep 2021 16:28:02 +0600 Subject: [PATCH] Update README.md --- README.md | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/README.md b/README.md index 17b89b1..10f4c41 100644 --- a/README.md +++ b/README.md @@ -53,16 +53,16 @@ For training and testing the U-net model, open the `train_unet_predictor.ipynb` ## Pre-trained models 1. Download the models from [google drive](https://drive.google.com/drive/folders/1XLehM5DYqLqiAsteO_h1PYZnavcCNOcR?usp=sharing). 2. Put the pre-trained models inside `/thermal-facial-landmarks-detection/models` directory. -3. dlib shape predictor -- To make predictions on images: +3. **dlib shape predictor** +- Make predictions on images: ``` python dlib_predict_image.py --images dataset/gray/test/images/ --models models/ --upsample 1 ``` -- To make predictions on a video: +- Make predictions on a video: ``` python dlib_predict_video.py --input video/2_0.avi --models models/ --upsample 1 --output video/output.mp4 ``` -4. U-net +4. **U-net model** ``` python unet_predict_image.py --dataset dataset/gray/test --model models/ ```