欢迎光临散文网 会员登陆 & 注册

海思hi3516dv300芯片AI图像模块学习(一)YOLO环境配置

2022-07-24 17:22 作者:饼干快快快跑  | 我要投稿

一、官方案例

https://gitee.com/openharmony/device_soc_hisilicon/tree/master/hi3516dv300/sdk_linux/sample/taurus#%E4%B8%80taurus%E7%9A%84%E7%9B%B8%E5%85%B3%E6%A6%82%E8%BF%B0

网页界面

直接从4.2开始

4.2 节

分类网里有流程图

流程

本来应该直接学4.2.4检测网络的yolo的,但是这个文档yolo用的darknet框架而且写的很简单,还是先学习分类网了解更常用的pytorch模型转caffe和wk的流程吧。

跟着下标注软件 https://www.gyan.dev/ffmpeg/builds/

选full版本

看了下ffmpeg是用来处理视频帧,matlab也能搞,没什么用跳过跳过。



二、下载yoloV5 pytorch 源码

好嘞 ,后面训练的用不到,跳转到yolo五的环境搭建和训练,方法参考Bubbliiiing大神的专栏

https://blog.csdn.net/weixin_44791964/article/details/121626848?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522165865340716782388070888%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fblog.%2522%257D&request_id=165865340716782388070888&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~blog~first_rank_ecpm_v1~rank_v31_ecpm-2-121626848-null-null.185^v2^control&utm_term=yolo%20v5&spm=1018.2226.3001.4450


源码在github上 :https://github.com/bubbliiiing/yolov5-pytorch

下载太慢了,

参考https://blog.csdn.net/weixin_44791964/article/details/123440488,把github代码转移到gitee上。

https://gitee.com/dashboard

新的url https://gitee.com/losschen/yolov5-pytorch.git

打开git bash程序,输入命令,git clone https://gitee.com/losschen/yolov5-pytorch.git

下载完成,打开requirement.txt 开始配环境。

三、搭建环境

找到Bubbliiiing 的yolov5环境为pytorch==1.2.0、torchvision==0.4.0

很好,开始配环境

视频教程:https://www.bilibili.com/video/BV1Ui4y147un;配置博客:https://blog.csdn.net/weixin_44791964/article/details/106037141 作者:Bubbliiiing https://www.bilibili.com/read/cv10239076?spm_id_from=333.999.0.0 出处:bilibili

然后我笔记本没有GPU ,使用CPU 的版本

https://blog.csdn.net/weixin_44791964/article/details/120655098?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522165865510916780357262525%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fblog.%2522%257D&request_id=165865510916780357262525&biz_id=0&utm_medium=distribute.pc_search_result.none-task-blog-2~blog~first_rank_ecpm_v1~rank_v31_ecpm-1-120655098-null-null.185^v2^control&utm_term=%E5%AD%A6%E4%B9%A0%E7%8E%AF%E5%A2%83%E9%85%8D%E7%BD%AE%20%20cpu&spm=1018.2226.3001.4450

3.1 卸载anaconda

以前装过anaconda了,由于很多配置忘记咋搞了,卸载重新装一下

https://blog.csdn.net/zlc2351951436/article/details/124590754?spm=1001.2101.3001.6661.1&utm_medium=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1-124590754-blog-95249601.pc_relevant_multi_platform_whitelistv3&depth_1-utm_source=distribute.pc_relevant_t0.none-task-blog-2%7Edefault%7ECTRLIST%7Edefault-1-124590754-blog-95249601.pc_relevant_multi_platform_whitelistv3&utm_relevant_index=1

参考这个卸载

conda install anaconda-clean
anaconda-clean --yes

搞定 conda info

3.安装anaconda

有vscode了,我选新版本

听说方便,勾上。

真慢

3.3 配置pytorch环境

cmd

Win+R启动cmd,在命令提示符内输入以下命令:

1.conda create –n pytorch python=3.6

前面一条指令用于创建一个名为pytorch的环境,该环境的python版本为3.6。

先下载py3.6

2.activate pytorch

后面一条指令用于激活一个名为pytorch的环境。

()里面应该表示进环境了

# CPU only

3.pip install torch==1.2.0+cpu torchvision==0.4.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

在我们新建的环境下载 torch==1.2.0+cpu torchvision==0.4.0

搞定

Successfully installed numpy-1.19.5 pillow-8.4.0 six-1.16.0 torch-1.2.0+cpu torchvision-0.4.0+cpu

4.装其他要求的库文件

 pip install -r C:\Users\Jack-Chen\Desktop\yolov5-pytorch\requirements.txt

这个在具体项目里有

scipy==1.2.1

numpy==1.17.0

matplotlib==3.1.2

opencv_python==4.1.2.30

torch==1.2.0

torchvision==0.4.0

tqdm==4.60.0

Pillow==8.2.0

h5py==2.10.0

一些常用库

慢的要死,吃饭

超级慢,安装完成环境也差不多了。先吃饭,吃早茶去

5.pip 换源

我都吃饭回来了还没下完。。


什么垃圾速度

老老实实换源,

需要注意的是,如果在pip中下载安装比较慢可以换个源,可以到用户文件夹下,创建一个pip文件夹,然后在pip文件夹里创建一个txt文件。

[global]

index-url = http://pypi.mirrors.ustc.edu.cn/simple

[install]

use-mirrors =true

mirrors =http://pypi.mirrors.ustc.edu.cn/simple/

trusted-host =pypi.mirrors.ustc.edu.cn

修改txt文件的内容,并且把后缀改成ini

用户文件夹pip

换了某个大学的源
300M网终于跑起来了

淦,这才对嘛,起飞!!

四、调教VScode

打开anaconda,切换环境。

在命令行里创建的pytorch环境已经可以看见啦,可以在这里切换环境。现在在VScode里还不能连上,下一步重启电脑。

重启成功,VS Code里也可以看到了。环境配完了,下一步运行模型测试一下。现在洗澡去~



海思hi3516dv300芯片AI图像模块学习(一)YOLO环境配置的评论 (共 条)

分享到微博请遵守国家法律