nvidia TAO有很多预训练模型,如: ##### Detection Network
- Faster-RCNN / YoloV3 / SSD / DSSD / RetinaNet / YoloV4 ?(https://ngc.nvidia.com/catalog/models/nvidia:tao:Faster-RCNN) - DetectNet_v2 (https://catalog.ngc.nvidia.com/orgs/nvidia/teams/tao/models/pretrained_detectnet_v2)
##### Classification Network
- multi_task (https://ngc.nvidia.com/catalog/models/nvidia:tao:pretrained_image_classification)
##### Other Networks
- DashCamNet (https://ngc.nvidia.com/catalog/models/nvidia:tao:dashcamnet) - VehicleMakeNet (https://ngc.nvidia.com/catalog/models/nvidia:tao:vehiclemakenet) - VehicleTypeNet (https://ngc.nvidia.com/catalog/models/nvidia:tao:vehicletypenet) - TrafficeCamNet (https://ngc.nvidia.com/catalog/models/nvidia:tao:trafficcamnet) - PeopleNet (https://ngc.nvidia.com/catalog/models/nvidia:tao:peoplenet) - FaceDetectIR (https://ngc.nvidia.com/catalog/models/nvidia:tao:facedetectir) - PeopleSegNet (https://ngc.nvidia.com/catalog/models/nvidia:tao:peoplesegnet) - PeopleSemSegnet (https://ngc.nvidia.com/catalog/models/nvidia:tao:peoplesemsegnet) 可以把这些模型放到deepstream中测试,步骤如下: 1 获取配置文件 $ git clone https://github.com/NVIDIA-AI-IOT/deepstream_reference_apps.git $ cd deepstream_reference_apps/deepstream_app_tao_configs/ $ sudo cp -a * /opt/nvidia/deepstream/deepstream/samples/configs/tao_pretrained_models/
2 下载模型 $ sudo apt install -y wget zip $ cd /opt/nvidia/deepstream/deepstream/samples/configs/tao_pretrained_models/ $ sudo ./download_models.sh ? 3 运行Yolo ./deepstream-app ?-c ../../../../samples/configs/tao_pretrained_models/deepstream_app_source1_detection_models.txt ?
|