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

Models? #5

Closed
StableInquest opened this issue Nov 11, 2022 · 5 comments
Closed

Models? #5

StableInquest opened this issue Nov 11, 2022 · 5 comments

Comments

@StableInquest
Copy link

How can I load models from here? https://github.com/open-mmlab/mmcv/blob/master/mmcv/model_zoo/open_mmlab.json

Thanks!

@dustysys
Copy link
Owner

To load official MMDetection pretrained models, pick one out from this list.
image
For those labelled 'Object Detection', place them in /models/mmdet/bbox. For those labelled 'Instance Segmentation', place them in /models/mmdet/segm.

You will need to download the model (.pth) and the config file (.py). Rename the model file to have the same name as the config file, mirroring the default ddetailer models. To download config files, click to view them and right click the 'Raw' button -> save link as. Here I grab the mmdet Mask2Former instance segmentation model as an example.
image

The official MMDetection models tend to have additional config files required, listed at the top of the config file.
image
You will need to download those from the repo as well, scroll up from the model downloads to view the root path of these configs. Sometimes those config files will list even further config files you need from the mmdet repo.
image
Download any additional configs from the relative path in the repo. Modify the relative paths in a text editor so that they are just pointing to the current directory so you can just keep them in the models folder with your model.
image
After collecting all the dependency configs the model directory will look something like this:
image
Restart the webui for the model to appear in your list:
image

@StableInquest
Copy link
Author

Thanks so much, that's really helpful!

@syddharth
Copy link

where can one find the default_runtime.py ?

@dustysys
Copy link
Owner

@soysanti
Copy link

soysanti commented Feb 1, 2023

ohh this looks great, would there be a way to subtract the head (the B) and not just the face?
also when it detects several bodies, to be able to define on which body we want to put the mask
Thank you!

@wkpark wkpark mentioned this issue May 4, 2023
5 tasks
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

No branches or pull requests

4 participants