With a ResNet-50 backbone and a number of architectural modifications, this version provides better accuracy and performance.
Our implementation is based on the existing model from the TensorFlow models repository. The network was altered in order to improve accuracy and increase throughput. Changes include:
This model was trained using script available on NGC and in GitHub repo
The following datasets were used to train this model:
Performance numbers for this model are available in NGC