Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
akuzdeuov authored Sep 10, 2021
1 parent fe79321 commit 75f6915
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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/
```
Expand Down

0 comments on commit 75f6915

Please sign in to comment.