search

NVIDIA-AI-IOT / TAO-Toolkit-Whitepaper-use-cases

TAO best practices. How to adapt for a new domain, new classes, and generalize the model with a small dataset using Nvidia's TAO toolkit
MIT License
24 stars 10 forks source link

Use TAO can I modify the model structure by myself? #2

Open CPFelix opened 3 years ago

CPFelix commented 3 years ago

I have train yolov4 example successfully and thank you for your great work! But if the pretrained model you provided is not meet the need of my project,can I modify the net structure or something else by myself?Thanks!

monjha commented 3 years ago

HI CPFelix,

You can modify the classes, preprocessing augmentation, backbone model, etc. Please check the list of supported backbone models here: https://docs.nvidia.com/tao/tao-toolkit/text/open_images/index.html#open-images