I want to Test Retail Object Detection Models provided under NGC in TAO. I want to first use the models to run inference, and then fine tune the models with my custom data. However I am not clear of which configuration/spec files to use with each provided model, if they are EfficientDet or DINO models ( and which TAO version)
The following information id provided under the documentation. However this is about models v1.0. Is the documentation not updated after although new models have been released?
Network Architecture: EfficientDet, DINO-FAN_base
The documentation mentioned to use TAO Efficientdet-TF2 for fine tuning the model and has provided the configuration file for that
`
However description for the latest release is given as
DINO (DETR with Improved DeNoising Anchor Boxes) based object detection network to detect retail objects on a checkout counter.
And the latest released trainable model is tagged as dino_model_epoch=011.pth
So its not clear to me what configuration/ spec file needs to be used with this trainable model when running inference with this model/ or to use them as pre-trained model.
- Can you please specify which TAO model and specification file need to be used with new or old trainable object detection models released under NGC. And also point to TAO model under TAO documentation (Efficient-det or DINO)? Are they are TF or pytorch models?
- And which TAO version we should use with latest released version ( TAO 5.2 as specified in documentation or is it out of date as the latest releases are after that?)
4 posts - 2 participants