Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[docker] Add docker with demo file #8

Open
wants to merge 5 commits into
base: master
Choose a base branch
from

Conversation

aounleonardo
Copy link

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

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
@hhj1897
Copy link
Member

hhj1897 commented Dec 2, 2022

We tried to make the repo OS agnostic, but it seems the dockerisation will only work on Linux, right? Of course that's fine, but probably worth mentioning in the README. Also, I think it will be nice to use ibug.face_detection as the detector, as we tried put all our face analysis libs into the same namespace.

@aounleonardo
Copy link
Author

@hhj1897

OS: So far I've only tested on MacOS, but it should be the same as Docker can run on [Mac, Linux, Windows]. I'll test on my Windows machine soon and figure out if it's seamless too.
Face Detector: I agree, it's a bit tricky to install, especially programmatically in docker, so I left it for future commits.

Leonardo Aoun and others added 4 commits December 3, 2022 16:08
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.
@aounleonardo
Copy link
Author

Screenshot 2022-12-03 at 19 47 37

Screenshot 2022-12-03 at 19 48 06

Screenshot 2022-12-03 at 19 48 16

@aounleonardo
Copy link
Author

[x] Verify on Windows
[x] Add demo for Jupyter
[ ] Add gpu
[ ] Script the model
[ ] RetinaFace <-- this is the most urgent/useful now

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.

2 participants