Skip to content

[docker] Add docker with demo file#8

Open
aounleonardo wants to merge 5 commits into
ibug-group:masterfrom
aounleonardo:dockerization
Open

[docker] Add docker with demo file#8
aounleonardo wants to merge 5 commits into
ibug-group:masterfrom
aounleonardo:dockerization

Conversation

@aounleonardo
Copy link
Copy Markdown

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
Copy Markdown
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
Copy Markdown
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
Copy Markdown
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
Copy Markdown
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