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、效果图

相关推荐
l1t8 分钟前
DeepSeek辅助实现的DuckDB copy to自定义函数
数据库·c++·人工智能
老歌老听老掉牙11 分钟前
旋量理论:刚体运动的几何描述与机器人应用
python·算法·机器学习·机器人·旋量
狐5715 分钟前
2025-05-08-deepseek本地化部署
人工智能·大语言模型
亚马逊云开发者31 分钟前
准确率从 19% 提升至 95%!文本审核模型优化的三个阶段实践(下)
人工智能
盛寒32 分钟前
词法分析和词性标注 自然语言处理
人工智能·自然语言处理
计算机集成_33 分钟前
具身智能之人形机器人核心零部件介绍
人工智能·经验分享·机器人
新智元44 分钟前
苹果 OS 全家桶 12 年最狠升级!AI 入侵一切,唯独 Siri 没更
人工智能·openai
我是初九1 小时前
【李沐-动手学深度学习v2】1.Colab学习环境配置
人工智能·python·学习·colab
失败又激情的man1 小时前
python爬虫之数据存储
前端·数据库·python
一刀到底2111 小时前
Python 高级应用10:在python 大型项目中 FastAPI 和 Django 的相互配合
python·django·fastapi