Pretrained model for retinanet transfer learning

Accelerated Computing Intelligent Video Analytics TAO Toolkit
1

I have a custom dataset for person detection and i want to train the model with retinanet (resnet18), but i couldn’t find any retinanet_resnet18 pretrained weight for transfer learning . Any help will be highly appreciated .

3

It is available in ngc.
Please see tlt user guide.

1
13729×1007 285 KB

4

Thanks.
How to download the resnet18 pretrained model ?

5

wget https://api.ngc.nvidia.com/v2/models/nvidia/tlt_pretrained_object_detection/versions/resnet18/files/resnet_18.hdf5

6

How to download it by using following command ?

ngc registry model download-version <ORG/model_name:version> -dest <path_to_download_dir>

7

Please refer to the jupyter sample.

!ngc registry model download-version nvidia/tlt_pretrained_object_detection:resnet18 --dest $USER_EXPERIMENT_DIR/pretrained_resnet18