/yolov5_for_pytorch

Primary LanguagePythonApache License 2.0Apache-2.0

项目名称:

yolov5 for pytorch

项目介绍

华为官方样例 Yolov5_for_pytorch精简版

可直接调用转换好的om模型,测试data文件夹中的jpg图片, 输出结果保存为json格式文件。

环境

  • cann 5.0.4

  • atlas500 (ascend 310)小站

  • python3

  • yolov5s

使用方法

sh run.sh