五个评分标准:
TAO训练及后续流程:
用tao进行训练,首先要转换数据格式,有tensorflow和pytorch两个版本。
结合预训练模型,进行训练,可以使用评估工具评估效果,效果好可以剪枝模型。
剪枝模型重新训练,然后导出。
TAO/data/box/object_detection 为txt格式标签
训练文件:ssd.ipynb
路径上的差别:
五个评分标准:
TAO训练及后续流程:
用tao进行训练,首先要转换数据格式,有tensorflow和pytorch两个版本。
结合预训练模型,进行训练,可以使用评估工具评估效果,效果好可以剪枝模型。
剪枝模型重新训练,然后导出。
TAO/data/box/object_detection 为txt格式标签
训练文件:ssd.ipynb
路径上的差别: