Pretrained model for retinanet transfer learning

samjith888 · August 27, 2020, 2:33pm

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 .

Morganh · August 28, 2020, 2:04am

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

samjith888 · August 28, 2020, 3:45am

Thanks.
How to download the resnet18 pretrained model ?

Morganh · August 28, 2020, 4:15am

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

samjith888 · August 28, 2020, 4:44am

How to download it by using following command ?

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

Morganh · August 28, 2020, 4:55am

Please refer to the jupyter sample.

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