PyTorch使用

前言

系统环境:win10

使用Anaconda,Anaconda的安装自行百度。

conda 23.7.4

目录

前言

创建虚拟环境

1、查看当前有哪些虚拟环境

2、创建虚拟环境pytorch

3、激活及关闭pytorch虚拟环境

4、删除pytorch虚拟环境

使用yolov5测试

1、切换至yolov5目录下:

2、安装相关依赖

3、下载已训练好的数据集

4、分类检测测试

1、执行命令

2、运行结果

3、效果图


创建虚拟环境

启动Anaconda Prompt

1、查看当前有哪些虚拟环境

bash 复制代码
conda env list

2、创建虚拟环境pytorch

bash 复制代码
conda create -n pytorch python=3.8

3、激活及关闭pytorch虚拟环境

bash 复制代码
# 激活pytorch虚拟环境
conda activate pytorch
# 关闭pytorch虚拟环境
conda deactivate

4、删除pytorch虚拟环境

bash 复制代码
conda remove -n pytorch --all

使用yolov5测试

yolov5下载地址:https://github.com/ultralytics/yolov5

1、切换至yolov5目录下:

bash 复制代码
# 切换至yolov5目录下
cd C:\code\python\lpytoch\yolov5

2、安装相关依赖

该下载过程时间比较长

bash 复制代码
# install
​​​​​​​pip install -r requirements.txt 

3、下载已训练好的数据集

直接下载

通过代码下载,代码保存为loadPyTorchHub.py

python 复制代码
# 使用YOLOv5 加载PyTorch Hub
# 简单示例
# 该示例从PyTorch Hub 加载预训练的 YOLOv5s 模型,即 model 并传递图像以供推理。 'yolov5s' 是最轻、最快的YOLOv5 型号
import torch

# Model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')

# Image
im = './data/images/zidane.jpg'

# Inference
results = model(im)

results.pandas().xyxy[0]
#      xmin    ymin    xmax   ymax  confidence  class    name
# 0  749.50   43.50  1148.0  704.5    0.874023      0  person
# 1  433.50  433.50   517.5  714.5    0.687988     27     tie
# 2  114.75  195.75  1095.0  708.0    0.624512      0  person
# 3  986.00  304.00  1028.0  420.0    0.286865     27     tie

结果如下:

4、分类检测测试

1、执行命令

bash 复制代码
# Inference with detect.py
python detect.py --weights yolov5s.pt --source ./data/images/zidane.jpg

2、运行结果

输出结果路径

Results saved to runs\detect\exp

3、效果图

相关推荐
xiao5kou4chang6kai47 分钟前
Python-GEE遥感云大数据分析与可视化(如何建立基于云计算的森林监测预警系统)
python·数据分析·云计算·森林监测·森林管理
铭keny13 分钟前
YOLO11 目标检测从安装到实战
人工智能·目标检测·目标跟踪
presenttttt14 分钟前
用Python和OpenCV从零搭建一个完整的双目视觉系统(四)
开发语言·python·opencv·计算机视觉
木头左3 小时前
逻辑回归的Python实现与优化
python·算法·逻辑回归
quant_19864 小时前
R语言如何接入实时行情接口
开发语言·经验分享·笔记·python·websocket·金融·r语言
杨小扩6 小时前
第4章:实战项目一 打造你的第一个AI知识库问答机器人 (RAG)
人工智能·机器人
whaosoft-1436 小时前
51c~目标检测~合集4
人工智能
雪兽软件6 小时前
2025 年网络安全与人工智能发展趋势
人工智能·安全·web安全
元宇宙时间7 小时前
全球发展币GDEV:从中国出发,走向全球的数字发展合作蓝图
大数据·人工智能·去中心化·区块链
小黄人20257 小时前
自动驾驶安全技术的演进与NVIDIA的创新实践
人工智能·安全·自动驾驶