Commit 67996fa3 authored by Prieur Maxime's avatar Prieur Maxime 🎵

Update README.md

parent 70432e21
......@@ -35,6 +35,11 @@ python yolo_video.py [video_path] [output_path (optional)]</code>
- You might have to modify the code to analyse custom video...
## Possible improvements
- The architecture is only a keras adaptation of the Darknet architecture and use the original with transfer learning might be more efficient.
- A lot of new architecture have been released since YOLOV3 and could provide better results. It would be interesting to train model like CenterNet ResNet-18 or DSOD300 + VoVNet-27-slim.
- Only a few pictures has been labeled and with not that much diversity on the content. Create a larger dataset would make a more robust model.
## Thanks
We would like to thanks the INSA of Rouen and Clément Chatelain our Machine-Learning
teacher for giving us the time and opportunity to work on a project on our own.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment