Comments (12)
sorry,I solved the problem,mainly for anchors and data labels,get the same score with author,Thank you.
from smart_construction.
same question,P is OK,but recall is low ,map0.5≈0.4
from smart_construction.
将 test.py
入参的默认值 conf_thres=0.4
,改大一点,我测试的时候将其设置为 0.6
from smart_construction.
将
test.py
入参的默认值conf_thres=0.4
,改大一点,我测试的时候将其设置为0.6
将你提供的helment_yolov5s.pt权重文件拿来测试,conf=0.6情况下,R还是只有0.4,还有其他地方需要修改吗
from smart_construction.
将
test.py
入参的默认值conf_thres=0.4
,改大一点,我测试的时候将其设置为0.6
阈值越大 召回就更低了呀
from smart_construction.
labels.npg里类别显示均衡吗,这边显示很不均衡
from smart_construction.
@PeterH0323
whether coordinate calculation error to bbox of head, helmet?
from smart_construction.
@PeterH0323
146th epoch:
people bbox P and R were better,while P and R of head or helmet were worse.
as following:
1.training:
145/299 9.17G 0.04331 0.1072 0.004936 0.1554 7980 640: 99%|███████████████████████▋| 85/86 145/299 9.17G 0.04325 0.1071 0.004934 0.1552 1441 640: 99%|███████████████████████▋| 85/86 145/299 9.17G 0.04325 0.1071 0.004934 0.1552 1441 640: 100%|████████████████████████| 86/86 145/299 9.17G 0.04325 0.1071 0.004934 0.1552 1441 640: 100%|████████████████████████| 86/86 [00:49<00:00, 1.74it/s]
Class Images Targets P R [email protected] [email protected]:.95: 10%|█▎ | 1/10 [00:03<00:32, 3.63s/i Class Images Targets P R [email protected] [email protected]:.95: 20%|██▌ | 2/10 [00:04<00:23, 2.92s/ Class Images Targets P R [email protected] [email protected]:.95: 30%|███▉ | 3/10 [00:06<00:17, 2.55s Class Images Targets P R [email protected] [email protected]:.95: 40%|█████▏ | 4/10 [00:07<00:12, 2.0 Class Images Targets P R [email protected] [email protected]:.95: 50%|██████▌ | 5/10 [00:08<00:08, 1. Class Images Targets P R [email protected] [email protected]:.95: 60%|███████▊ | 6/10 [00:08<00:05, 1 Class Images Targets P R [email protected] [email protected]:.95: 70%|█████████ | 7/10 [00:09<00:03, Class Images Targets P R [email protected] [email protected]:.95: 80%|██████████▍ | 8/10 [00:09<00:01, Class Images Targets P R [email protected] [email protected]:.95: 90%|███████████▋ | 9/10 [00:10<00:00 Class Images Targets P R [email protected] [email protected]:.95: 100%|████████████| 10/10 [00:10<00:00 Class Images Targets P R [email protected] [email protected]:.95: 100%|████████████| 10/10 [00:10<00:00, 1.05s/it]
2.verification:
Class Images Targets P R [email protected] [email protected]:.95
all 607 7.38e+04 0.893 0.352 0.342 0.232
person 607 4.36e+03 0.846 0.798 0.776 0.536
head 607 6.42e+04 0.903 0.128 0.123 0.0635
helmet 607 5.23e+03 0.931 0.129 0.127 0.0965
from smart_construction.
sorry,I solved the problem,mainly for anchors and data labels,get the same score with author,Thank you.
你好,我也遇到了同样的问题,请问你的(yolov5s模型基础)的训练脚本的epoch是多少呢,还有你自己kmeans跑出来的anchor数值大小是和作者README一样么,非常感谢你的回答
from smart_construction.
我已更新文件 models/custom_data.yaml
,更新了我经过kmeans
的anchors
和网络的深度和宽度,请试一下
from smart_construction.
问题已经的到解决,先关闭这个 issue 了,后续如果有疑问的话欢迎继续提 issue ,感谢大家的关注和支持!
from smart_construction.
其实这个问题是正确使用文件:./data/gen_data/merge_data.py,因为在提取人体类别和坐标的时候会生成一次人体lable,如果操作多次,会重复生成人体lable。另一个原因是其它数据库的制作人体标签时,是不能和Safety-Helmet-Wearing-Dataset数据库合并的,而是单独标签。我迁移训练一轮效果:
Class Images Targets P R [email protected] [email protected]:.95: 100%
all 3.3e+03 2.07e+04 0.516 0.858 0.815 0.47
person 3.3e+03 1.08e+04 0.408 0.811 0.729 0.44
head 3.3e+03 9.18e+03 0.48 0.862 0.814 0.345
helmet 3.3e+03 747 0.658 0.902 0.902 0.624
这里Safety-Helmet-Wearing-Dataset数据是把train和test合在一起(当然单幅图标签是独立的),大概69k,用val数据(约0.6k)测试。coco train提取的人体标签约6万多点,这样共7.1万训练集,验证集共33k。
所以,作者的测试结果是真实可信的,我没有用custom_yolov5.yaml,只用了yolov5s.yaml,当然也没有./data/gen_anchors/clauculate_anchors.py,聚类得出先验框。
from smart_construction.
Related Issues (20)
- 请问支持RTSP实时视频推理吗? HOT 1
- onnx export can work HOT 2
- 关于打包exe问题 HOT 7
- 数据集 HOT 1
- 安全帽和非安全帽 HOT 1
- 截图保存问题 HOT 1
- 如何判断是安全帽是手拿着还是头戴着? HOT 1
- Can you put the weights somehwere else for easier download?
- 可视化界面引入实时监控问题 HOT 1
- CUDA error
- 想问下如何实现对本地摄像头的实时检测呢 HOT 1
- 如何解决摄像头远距离、俯瞰角度、小尺寸安全帽检测率低问题? HOT 1
- 执行train.py错误,请教解决 HOT 4
- 增加数据集的分类具体怎么操作?可以分享下增加人体一项的数据集吗? HOT 1
- Ubuntu 无法启动Ui界面!
- 这里给大家提几个小建议! HOT 1
- 您好为什么我的visual_interface.py一运行就卡死 HOT 1
- requirements.txt的mkl-fft,mkl-random, mkl-service的版本 HOT 3
- 安装问题 HOT 1
- 在运行visual_interface.py出现Error: "No valid frames decoded before end of stream" HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from smart_construction.