docker login nvcr.io Username: $oauthtoken Password: aXE2azBiZG90N2JzNTRvdGt1NnR1OHJpZ2o6NTQwZjQyNjYtMDlhNS00YWQ0LWFmMzYtMmM5ZGM3ZmZjYjM4
API Key: aXE2azBiZG90N2JzNTRvdGt1NnR1OHJpZ2o6NTQwZjQyNjYtMDlhNS00YWQ0LWFmMzYtMmM5ZGM3ZmZjYjM4
forums.developer.nvidia.com/accelerated-computing/intelligent-video-analytics/tao-toolkit/17
首先要配置训练所需要的环境。例如Tao
准备数据集和预训练模型,其中预训练模型可以选择有很多,都可以在NGC上下载。
下载预训练模型后,设置训练参数
利用TAO开始训练
训练结束后评估训练结果
挑选mAP最好的模型进行模型剪枝
对剪枝后的模型重新训练
对剪枝后的模型进行可视化推理结果
把模型导出,然后部署到NANO上