Skip to content

Add retinaface to Docker demo#9

Open
aounleonardo wants to merge 7 commits intoibug-group:masterfrom
aounleonardo:add_retina
Open

Add retinaface to Docker demo#9
aounleonardo wants to merge 7 commits intoibug-group:masterfrom
aounleonardo:add_retina

Conversation

@aounleonardo
Copy link
Copy Markdown

/!\ Don't accept before #8

Screenshot 2022-12-04 at 00 21 35

Leonardo Aoun and others added 7 commits December 2, 2022 18:14
Add a dockerfile which installs the prereqs.
When built and ran, User will automatically enter a tmux session with FAN predicting the Astronaut sklearn image
When built and ran, User will automatically enter a tmux session with FAN predicting the Astronaut sklearn image, this behaviour can be changed in future commits.

Detector used for now is simply the HaarCascade from opencv.

Next steps:
- Either build a flask server with UploadImage button, detects the face and draws the landmarks
- Or a jupyter notebook with a similar functionality
- Script the model, currently I cannot do it because _modules is an issue in HourGlass
- I haven't yet tested on GPUs, I believe I need to detect the gpu in bash and pass them using --gpus all
Still not able to run on GPU but it's not a huge issue as demoing seems fine on a CPU, it takes .3 secs per image.
Might not work for all versions of windows or environments, but I guess it's a working progress.

Next steps:
- Add the jupyter notebook demo
- Figure out how to pull RetinaFace.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant