NVIDIA
NVIDIA
SSD v1.1 for PyTorch
Resource
NVIDIA
NVIDIA
SSD v1.1 for PyTorch

With a ResNet-50 backbone and a number of architectural modifications, this version provides better accuracy and performance.

Setup

The following section list the requirements in order to start training the SSD300 v1.1 model.

Requirements

This repository contains Dockerfile which extends the PyTorch 19.03 NGC container and encapsulates some dependencies. Aside from these dependencies, ensure you have the following software:

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: