【2020华为云AI实战营】基于MA Jupyter 实现 yoloV5 目标检测（二）

Yolov5

一、环境准备

【2020华为云AI实战营】Modelarts Notebook中搭建python3.7+pytorch1.5环境 （一）

https://bbs.huaweicloud.com/blogs/180972

二、代码拉取及运行

yoloV5 github地址：https://github.com/ultralytics/YOLOv5

（1）拉取代码：如下：!git clone https://github.com/ultralytics/yolov5  # clone repo

（2）打开Terminal并且进入文件目录下：

        cd work/yolov5/

（3）激活环境py37：

        activate py37

（4）运行代码：

        python detect.py --weights yolov5s.pt --img 416 --conf 0.4 --source ./inference/images/

（5）进入inference/output目录下查看运行结果：

三、其他

yolov5推理速度很快可以应用到移动端实现

python detect.py --source file.jpg  # image 
                            file.mp4  # video
                            ./dir  # directory
                            0  # webcam
                            rtsp://170.93.143.139/rtplive/470011e600ef003a004ee33696235daa  # rtsp stream
                            http://112.50.243.8/PLTV/88888888/224/3221225900/1.m3u8  # http stream

四、建议

后续如果MA推出将自定义配置的环境保存下来，每次启动可以将之前配置好的环境在启动容器时可以自动恢复，并且在界面出现对应的按钮就好了。

以上。