Pytorch学习笔记（六）

于 2022-12-06 15:34:55 发布
阅读量526 收藏
点赞数
分类专栏： 深度学习 文章标签： pytorch 学习 python 深度学习
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/weixin_56744177/article/details/128204007
版权

目录

01 配置环境

①Conda环境

②项目环境

02 利用YOLOv5预测

03 训练YOLOv5模型

本笔记主要记录了以github上的开源项目为例，利用YOLOv5进行目标检测、训练YOLOv5神经网络、制作和训练自己的数据集的学习过程。

项目地址：https://github.com/ultralytics/yolov5

项目版本：Tags=v6.0

01 配置环境

①Conda环境

下载github项目后，在pycharm中打开文件，并在setting中设置解释器为之前配置好的pytorch环境。

②项目环境

一般项目所需环境会写在requirements.txt文件中。可以利用Pycharm自带的智能提示进行安装；也可以直接在终端台利用pip install -r requirements.txt指令安装。

如果作者没有提供requirements.txt文件，可以根据运行报错信息百度，手动安装缺少的库。

02 利用YOLOv5预测

利用作者已经训练好的YOLOv5对自定义输入数据进行预测。主要是在detect.py文件中操作。即可操作包括添加自定义输入数据、更改参数、运行detect.py文件。

detect.py参数含义

使用默认参数取值运行detect.py结果

更改输入数据路径（利用annie软件下载的b站视频）

更改输入视频运行结果

03 训练YOLOv5模型

主要是在train.py文件中操作。通过更改合适的参数来实现本地训练。

由于本机训练过程中总是出错、Pycharm闪退，百度搜索解决方案后仍未解决，故无法展示运行结果。

train.py参数含义

Hygge0+
关注
专栏目录
pytorch学习笔记六：torch.utils.data下的TensorDataset和DataLoader的使用
耐心的小黑的博客
04-08 1万+
一、TensorDataset 对给定的tensor数据(样本和标签)，将它们包装成dataset ''' data_tensor (Tensor) －　样本数据 target_tensor (Tensor) －　样本目标（标签） ''' dataset=torch.utils.data.TensorDataset(data_tensor, ...
matlab麦克斯韦速率代码-pytorch-detect-to-track:检测和跟踪的pytorch实现（https://arxiv.or
05-20
matlab麦克斯韦评分代码本文的pytorch实现。 介绍 该项目是检测跟踪和跟踪检测的pytorch实现。 该存储库受以下实现的影响： ，基于Pytorch ，基于Pycaffe + Numpy ，基于Pytorch + Numpy ，基于TensorFlow + Numpy ，Pytorch + TensorFlow + numpy 在我们的实现过程中，我们特别参考上面的实现。 在该实现中，此存储库具有以下品质： 它是纯Pytorch代码。 我们将所有的numpy实现都转换为pytorch！ 它支持多图像批处理训练。 我们修改了所有层，包括数据加载器，rpn，roi池等，以在每个minibatch中支持多个图像。 它支持多个GPU训练。 我们使用了多个GPU包装器（此处为nn.DataParallel），可以灵活地使用一个或多个GPU，这是上述两个功能的优点。 此外，由于“检测到跟踪和跟踪到检测”实现最初使用R-FCN暹罗网络和相关层，因此我们添加/修改了以下内容： 每个roidb条目支持多个图像。 默认情况下，我们在连续帧中使用2张图像来定义一个roidb条目，以方便通过两足暹
pytorch 实时物体检测
Yonggie的博客
12-20 1619
前言 使用了pytorch自己提供的maskrcnn_resnet50_fpn，也就是说，不过是调用人家训练好的代码而已，当个玩具了，供大家参考。 github项目 其实主要的代码量在结果的显示上，并不是模型的调用上。 这个是github项目链接. 效果： 实现思路 模型调用，这个就不必细说了。 使用opencv读图片 然后转化成torch能吃进去的数据， input到模型中，得到返回值。根据模型的返回值，一一把label标注到原图片中 然后显示图片 下面的是主要代码： 不能直接运行，只是思路，完整的
YOLOv5(PyTorch)目标检测：原理与源码解析
12-07
【为什么要学习这门课】 Linux创始人Linus Torvalds有一句名言：Talk is cheap. Show me the code. 冗谈不够，放码过来！代码阅读是从基础到提高的必由之路。 YOLOv5是最近推出的轻量且高性能的实时目标检测方法。YOLOv5使用PyTorch实现，含有很多业界前沿和常用的技巧，可以作为很好的代码阅读案例，让我们深入探究其实现原理，其中不少知识点的代码可以作为相关项目的借鉴。 【课程内容与收获】 本课程将详细解析YOLOv5的实现原理和源码，对关键代码使用PyCharm的debug模式逐行分析解读。 本课程将提供注释后的YOLOv5的源码程序文件。   【相关课程】 本人推出了有关YOLOv5目标检测的系列课程。请持续关注该系列的其它视频课程，包括：《YOLOv5(PyTorch)目标检测实战：训练自己的数据集》Ubuntu系统 https://edu.csdn.net/course/detail/30793 Windows系统 https://edu.csdn.net/course/detail/30923 《YOLOv5(PyTorch)目标检测：原理与源码解析》https://edu.csdn.net/course/detail/31428 《YOLOv5(PyTorch)目标检测实战：Flask Web部署》https://edu.csdn.net/course/detail/31087 《YOLOv5(PyTorch)目标检测实战：TensorRT加速部署》https://edu.csdn.net/course/detail/32303
pytorch——yolov3 detect部分学习记录
BigDiorXixi的博客
11-22 453
关于yolov3中detect即目标检测部分的一些学习中做的注释 代码来自：https://github.com/eriklindernoren/PyTorch-YOLOv3 from __future__ import division from models import * from utils.utils import * from utils.datasets import * import os import sys import time import datetime import
【PyTorch】目标检测与场景分割神器 - Detectron2
北境の守望者
04-30 4389
Backto PyTorch Index Detectron2 @ 2019-10-10 github $$ docs Detectron2 ，对 Detectron1 彻底重写，抛弃 Caffe2，直接基于 PyTorch （>= 1.3）, 以 Mask-RCNN benchmark 为基底，全新构建。 基于 PyTorch：直观，友好 模块化：灵活，可扩展，能够直接在单个或多个 ...
pytorch学习笔记（一）
01-06
此为小弟pytorch的学习笔记，希望自己可以坚持下去。（2020/2/17） pytorch官方文档 pytorch中文教程 tensor tensor是pytorch的最基本数据类型，相当于numpy中的ndarray，并且属性和numpy相似，tensor可在GPU上进行...
笔记pytorch学习笔记
07-20
【PyTorch学习笔记概述】 PyTorch是Facebook开源的一款深度学习框架，它以其灵活性、易用性和强大的计算能力在学术界和工业界都受到了广泛的欢迎。本笔记将基于B站牛二大人的讲解，深入探讨PyTorch的核心概念、基本...
pytorch学习笔记
最新发布
07-15
pytorch学习笔记
pytorch 学习笔记
06-27
pytorch 学习笔记
用Retinaface_pytorch项目检测人脸+ Python 3 剪切人脸图片并保存
weixin_45654744的博客
04-13 3749
概述 因为我的项目中缺一不可的一部分是提取人脸特征，因为以前并没有接触过人脸的项目，所以想把自己每次的一点点进步和学习都记录下来，这次是记录我学习切割人脸图片并保存，以及将人脸检测框和五个特征点坐标信息保存在.txt文件中。 这篇文档的方法要稍微复杂一点点，我结合Retinaface_pytorch项目中的detect.py（检测单张图片）来检测人脸，并通过以下方法成功的剪切出人脸，并保存特征点信...
anaconda安装pytorch 与 detectron安装
qq_21399393的博客
07-21 1903
TIP 目前pytorch已有1.0版本，直接pip安装即可！很方便！ 以下是从源码编译安装！ 绪论 最近在做一些目标检测的问题，想使用一些目标检测的API。之前使用的是tensorflow的object detection API，效果一般。据大佬汇报，detectron是一个很好的目标检测的API，而且方便使用。 我就想尝试安装测试一下，安装过程中，参考了很多博客...
用PyTorch来做物体检测和追踪(复现:pytorch_objectdetecttrack-master)
jiruijing123的博客
03-25 3229
用PyTorch来做物体检测和追踪 原文:https://towardsdatascience.com/object-detection-and-tracking-in-pytorch-b3cf1a696a98 翻译:https://ai.yanxishe.com/page/TextTranslation/1333 源码:https://github.com/cfotache/pytorch_ob...
5. 使用PyTorch预先训练的模型执行目标检测
qq_40985985的博客
06-28 3889
这篇博客将介绍如何使用PyTorch预训练的网络执行目标检测，这些网络是开创性的、最先进的图像分类网络，包括使用ResNet的更快R-CNN、使用MobileNet的更快R-CNN和RetinaNet。 - 具有ResNet50主干的更快R-CNN（更精确，但速度较慢） - 具有MobileNet主干的更快R-CNN（速度更快，但准确度稍低） - 具有ResNet50主干的RetinaNet（速度和准确性之间的良好平衡）............
GitHub趋势榜第一：超强PyTorch目标检测库Detectron2，训练更快，支持更多任务
量子位
10-16 1511
栗子 发自 凹非寺量子位 报道 | 公众号 QbitAIPyTorch目标检测库Detectron2诞生了，Facebook出品。站在初代的肩膀上，它训练比从前更快，功能...
1 PyTorch版YOLOv3 代码中文注释 之 训练 train.py test.py detect.py
轮子去哪儿了
08-12 7576
文章目录PyTorch版YOLOv3代码中文注释1. 相关链接：2. 文件结构：3. train.py3.1. train.py 中包含的主要功能3.2. train.py 每行代码添加中文注释3.3. train.py 涉及的重要函数、知识点3.3.1. argparse 库的使用方法3.3.2. Logger 库的使用？？不会3.3.3. device = torch.device（） 设置...
detectron_pytorch_code_reading
ayst123 的专栏
05-10 596
PyTorch-YOLOv3中detect.py 出现的输入问题
weixin_43590151的博客
10-05 1672
博主在运行yolov3中遇见 RuntimeError: Given groups=1, weight of size 32 3 3 3, expected input[1, 1, 416, 416] to have 3 channels, but got 1 channels instead。 现解决方法为： detect.py使用utils.datasets.py中的ImageFolder类将...
【pytorch1.3+detectron2】安装记录
Watson的博客
11-22 2883
官方安装就很好 https://github.com/facebookresearch/detectron2/blob/master/INSTALL.md 0. 安装Cuda9.0+cuDNN7.4 1. anaconda创建虚拟环境：conda create -n pytorch13 python=3.6.8，进入环境，conda activate pytorch13 2. 安装pytorch1...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值