|
@@ -1,15 +1,15 @@
|
|
|
# cambricon_pytorch
|
|
# cambricon_pytorch
|
|
|
寒武纪mlu220 转模型代码：yolov5-6.0_v2.zip
|
|
寒武纪mlu220 转模型代码：yolov5-6.0_v2.zip
|
|
|
|
|
|
|
|
-step1、detect.py 验证
|
|
|
|
|
|
|
+step1、detect.py 验证:
|
|
|
|
|
|
|
|
python detect.py --device cpu --weights yolov5s.pt
|
|
python detect.py --device cpu --weights yolov5s.pt
|
|
|
|
|
|
|
|
-step2、detect_qua.py 量化
|
|
|
|
|
|
|
+step2、detect_qua.py 量化:
|
|
|
|
|
|
|
|
python detect_qua.py --weights yolov5s.pt --cfg qua
|
|
python detect_qua.py --weights yolov5s.pt --cfg qua
|
|
|
|
|
|
|
|
-step3、detect_qua_mlu.py 生成mlu220离线文件
|
|
|
|
|
|
|
+step3、detect_qua_mlu.py 生成mlu220离线文件:
|
|
|
|
|
|
|
|
python detect_qua_mlu.py --weights yolov5s.pt --cfg mlu --jit True
|
|
python detect_qua_mlu.py --weights yolov5s.pt --cfg mlu --jit True
|
|
|
|
|
|
