Resource
With a ResNet-50 backbone and a number of architectural modifications, this version provides better accuracy and performance.
Use the NGC CLI to download:
Copied!
Setup
The following section list the requirements in order to start training the SSD320 v1.2 model.
Requirements
This repository contains Dockerfile which extends the TensorFlow NGC container and encapsulates some dependencies. Aside from these dependencies, ensure you have the following software:
- NVIDIA Docker
- TensorFlow 19.03-py3 (or later) NGC container
- NVIDIA Volta based GPU
For more information about how to get started with NGC containers, see the following sections from the NVIDIA GPU Cloud Documentation and the Deep Learning Documentation: