当前位置:网站首页>工程(五)——小目标检测tph-yolov5
工程(五)——小目标检测tph-yolov5
2022-07-31 02:34:00 【桦树无泪】
1、环境安装
1、创建环境
conda create -n yolo python=3.7
conda activate yolo
2、安装pytorch
conda install pytorch torchvision cudatoolkit=11.3 -c pytorch
3、克隆代码
git clone https://github.com/cv516Buaa/tph-yolov5
cd tph-yolov5
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple -r requirements.txt
4、yolov5 报错AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor
根据自己电脑cuda执行相应命令
# CUDA 10.2
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch
# CUDA 11.3
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=11.3 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cpuonly -c pytorch
2、数据下载
Datasets: [VisDrone](http://aiskyeye.com/download/object-detection-2/)
Weights(PyTorchv1.10):
yolov5l-xs-1.pt: (https://pan.baidu.com/s/1APETgMoeCOvZi1GsBZERrg)
yolov5l-xs-2.pt: (https://pan.baidu.com/s/19S84EevP86yJIvnv9KYXDA)
3、测试
python detect.py --source ./data/images/ --weights weights/yolov5l-xs-1.pt --conf 0.4
边栏推荐
- Crypto Life, a day in the life of a Web3 project partner
- Software Testing Defect Reporting - Definition, Composition, Defect Lifecycle, Defect Tracking Post-Production Process, Defect Tracking Process, Purpose of Defect Tracking, Defect Management Tools
- 完整复制虚拟机原理(云计算)
- Linux下redis7的安装,启动与停止
- Live Preview | KDD2022 Doctoral Dissertation Award Champion and Runner-up Dialogue
- 开题报告之论文框架
- Calculate S=a+aa+…+aa…a
- LeetCode 1161 The largest element in the layer and the LeetCode road of [BFS binary tree] HERODING
- [1153] The boundary range of between in mysql
- Coldfusion file read holes (CVE - 2010-2861)
猜你喜欢
基于FPGA的售货机
Introduction and use of Drools WorkBench
JS 函数 this上下文 运行时点语法 圆括号 数组 IIFE 定时器 延时器 self.备份上下文 call apply
图像处理技术的心酸史
StringJoiner详解
f.grid_sample
基于FPGA的图像实时采集
Basic learning about Redis related content
The application of AI in the whole process of medical imaging equipment
What are the project management tools like MS Project
随机推荐
系统需求多变如何设计
MPPT太阳能充放电控制器数据采集-通过网关采集电池电压容量电量SOC,wifi传输
221. Largest Square
Brute Force/Adjacency List Breadth First Directed Weighted Graph Undirected Weighted Graph
tcp框架需要解决的问题
try-catch中含return
f.grid_sample
Intel's software and hardware optimization empowers Neusoft to accelerate the arrival of the era of smart medical care
[1153]mysql中between的边界范围
完整复制虚拟机原理(云计算)
Mathematical Ideas in AI
Layer 2 broadcast storm (cause + judgment + solution)
What are the project management tools like MS Project
To write good test cases, you must first learn test design
The modification is not properly placed in the sandbox, causing Apple compatibility issues
【AcWing 62nd Weekly Game】
Force buckled brush the stairs (7/30)
Installation, start and stop of redis7 under Linux
Verify the integer input
局域网电脑硬件信息收集工具