首页 > 其他分享 >pytorch yolov3 推理和训练环境搭建

pytorch yolov3 推理和训练环境搭建

时间:2022-12-05 12:06:31浏览次数:69  
标签:detect yolov3 -- jpg pytorch 搭建 推理 分支


本文基于anaconda python3.8.5环境搭建

pytorch yolov3 推理和训练环境搭建_开发环境

关于环境的安装参考


下载环境

首先下载pytorch yolov3的推理和训练开发环境

git clone https://github.com/ultralytics/yolov3

pytorch yolov3 推理和训练环境搭建_人工智能_02

我们在默认的主分支上耍

pytorch yolov3 推理和训练环境搭建_人工智能_03

安装依赖包 

执行:

pip install -r ./requirements.txt

pytorch yolov3 推理和训练环境搭建_深度学习_04

有几个比较大的包,比如torch,opencv等需要花比较长的时间安装。

根据仓库提供的介绍, ​​ultralytics/yolov3​​ 仓库包含两个主要分支,master分支前向兼容yolov5模型算法,是它官方推荐的使用分支,而另一个archive分支后向兼容老的backward darknet 格式的模型文件,并且此分支不在维护,所以不推荐使用。

既然如此,我们使用master分支。

推理

目录中的​​detect.py​​​ 执行推理过程,并将结果保存在​​runs/detect目录下​

输入命令

python detect.py --source data/images --weights yolov3.pt --conf 0.25

进行推理,此命令首先会从以下地址下载权重文件

pytorch yolov3 推理和训练环境搭建_开发环境_05

https://github.com/ultralytics/yolov3/releases/download/v9.5.0/yolov3.pt

pytorch yolov3 推理和训练环境搭建_人工智能_06

 如果环境本身的下载过程比较慢,你也可以用迅雷等快一些的下载工具下载过来,在放入目录,此后执行命令检测到文件存在,则会PASS调下载环节。

命令成功执行后,显示如下:

pytorch yolov3 推理和训练环境搭建_python_07

 根据输出,我们知道模型从bus.jpg和zidane.jpg中得到了检测结果

image 1/2 /home/caozilong/Workspace/yolov3/data/images/bus.jpg: 640x480 4 persons, 1 bus, Done. (0.490s)
image 2/2 /home/caozilong/Workspace/yolov3/data/images/zidane.jpg: 384x640 2 persons, 3 ties, Done. (0.385s)

并将检测结果放在了run/detect/exp18目录下

pytorch yolov3 推理和训练环境搭建_开发环境_08

pytorch yolov3 推理和训练环境搭建_python_09

训练:

执行命令

python train.py --data coco.yaml --cfg yolov3.yaml --weights '' --batch-size 24

开始进行训练

pytorch yolov3 推理和训练环境搭建_开发环境_10

pytorch yolov3 推理和训练环境搭建_pytorch_11

这是在另一个PC下搭建环境的推理结果:

对比上面两幅图可以看出,他们的推理结论完全相同,这说明同一个模型在同样的权重配置下,结果是确定的。


结束

标签:detect,yolov3,--,jpg,pytorch,搭建,推理,分支
From: https://blog.51cto.com/u_15899439/5911827

相关文章

  • pytorch中调整学习率的方法
    1.lr_scheduler.StepLRclasstorch.optim.lr_scheduler.StepLR(optimizer,step_size,gamma=0.1,last_epoch=-1)功能: 等间隔调整学习率,调整倍数为gamma倍,调整间隔为st......
  • 1.4 Apache Hadoop完全分布式集群搭建-hadoop-最全最完整的保姆级的java大数据学习资
    目录1.4ApacheHadoop完全分布式集群搭建1.4.1虚拟机环境准备1.4.2集群规划1.4.3安装Hadoop1.4.3.1集群配置1.4.3.1.1HDFS集群配置1.4.3.1.2MapReduce集群配置1.4.......
  • 如何搭建一个自己的音乐服务器
    点赞再看,动力无限。微信搜「程序猿阿朗」。本文Github.com/niumoo/JavaNotes和未读代码博客已经收录,有很多知识点和系列文章。最近发现,经常用的网易云音乐,有很......
  • 阿里云oss的服务搭建
    最近自己在做项目,涉及到文件存储的需求。之前在公司的时候用的是公司的自己搭建的文件服务器,考虑到现在需要快速开发我的项目,先用阿里云的oos服务。1.快速。......
  • 3.搭建Linux环境
    Linux在学习编程的过程中非常重要,要学习Linux那就得先搭建Linux环境,本文将介绍如何在云服务器上搭建环境。Linux环境的搭建方式主要有三种:直接安装在物理机上,但是由于L......
  • 搭建WordPress个人博客
    简介WordPress是一个以PHP和MySQL为平台的自由开源的博客软件和内容管理系统。WordPress具有插件架构和模板系统。截至2018年4月,排名前1000万的网站中超过30.6%使用Word......
  • pytorch安装
    pytorch安装1、查看本机的CUDA版本cmd命令行输入nvidia-smi,在第一行最右边可以看到CUDA的版本号![version](C:\Users\nice7\Pictures\SavedPictures\version.png)2、......
  • 在本地搭建 SVN 教程
    SVN使用教程以下内容参考自:SVN使用详细教程_大梦谁先觉i的博客-CSDN博客_svn使用教程一、SVN安装1.1软件下载服务器:Downloads|VisualSVN客户端:Downloads·To......
  • go环境搭建及tjfoc-gm安装
    go环境搭建及tjfoc-gm安装一、环境配置1首先进入usr/local文件夹,下载go语言环境配置压缩包wgethttps://dl.google.com/go/go1.19.3.linux-amd64.tar.gz2配置......
  • Android-开发环境搭建02
    01Android开发工具AndroidStudioAndroidStudio是谷歌推出的一个Android集成开发工具,基于IntelliJIDEA.类似EclipseADT,AndroidStudio提供了集成的Android开......