Skip to content

Commit

Permalink
Add deep learning notebook and update readme with correct pretrained …
Browse files Browse the repository at this point in the history
…networks
  • Loading branch information
Cyril9227 committed Dec 29, 2020
1 parent 5b0d536 commit d350097
Show file tree
Hide file tree
Showing 2 changed files with 7,582 additions and 6 deletions.
16 changes: 10 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -69,10 +69,14 @@ The app will first download and instantiate the model when the API is launched t
| MobileNetV2-FPN | 98.3 ms | 95 | 62 |
| VoVNet-19-FPN | 95.6 ms | 90 | 85 |

Final trained weights for each model are available [here](https://www.dropbox.com/sh/yyqecdmbd7howkq/AADxOHGQQXtk7Jn8nIrTpG0Ha?dl=0) :
- ResNet-50-FPN : https://www.dropbox.com/s/we9n6mr1uyxlqt9/ResNet_50_FPN_model_final.pth?dl=1
- ResNet-101-FPN : https://www.dropbox.com/s/7xtyuitiembeu79/ResNet101_FPN_model_final.pth?dl=1
- MobileNetV2-FPN : https://www.dropbox.com/s/5ndg2gl153nap2x/MobileNetV2_FPN_model_final.pth?dl=1
- VoVNet-FPN : https://www.dropbox.com/s/01879hbpditrv13/VoVNet_lite_FPN_model_final.pth?dl=1
- Final trained weights for each model are available on dropbox :
- Download links for each model can be used directly as a config entry or as a script argument, those links are provided in the inference_config.
- ResNet-50-FPN : https://www.dropbox.com/s/yn7m8xnva068glq/ResNet50_FPN_model_final.pth?dl=1
- ResNet-101-FPN : https://www.dropbox.com/s/otp52ccygc2t3or/ResNet101_FPN_model_final.pth?dl=1
- MobileNetV2-FPN : https://www.dropbox.com/s/tn6fhy829ckp5ar/MobileNetV2_FPN_model_final.pth?dl=1
- VoVNet-FPN : https://www.dropbox.com/s/smm7t8jsyp05m4r/VoVNet19_FPN_model_final.pth?dl=1

*Nb : Download links for each model can be used directly as a config entry or as a script argument, those links are provided in the inference_config.*

- Visual outputs for each model are available under `reports/figures/model_name/predicted_images/`

- Training metrics for each model are available under `models/model_name/`
Loading

0 comments on commit d350097

Please sign in to comment.