bubbliiiing / yolov7-pytorch

这是一个yolov7的库,可以用于训练自己的数据集。
GNU General Public License v3.0
861 stars 150 forks source link

用predict.py测试fps才30左右 #29

Open charming2992 opened 2 years ago

charming2992 commented 2 years ago

自己训练了voc数据集,测试fps大概在30左右。 用get_map测试mAP到90多了

charming2992 commented 2 years ago

用的3090的卡

bubbliiiing commented 2 years ago

环境是?

charming2992 commented 2 years ago

环境是?

环境是指的什么呀? 用了官方的v7训练了VOC数据集之后测试的速度如下(按照这个total time是100左右的fps) Speed: 8.9/0.8/9.7 ms inference/NMS/total per 640x640 image at batch-size 1

charming2992 commented 2 years ago

环境是?

torch是1.9.0

bubbliiiing commented 2 years ago

可以描述一下你看的哪里显示的fps吗? 是get_map的吗? 我这里笔记本的1660ti使用predict.py里面的fps是18.67

bubbliiiing commented 2 years ago

如果看的是get_map的,那个是不准的。正常不会把门限设置成0.001

charming2992 commented 1 year ago

就是在predict.py里面选择mode为fps测得呀

bubbliiiing commented 1 year ago

你能去get fps里面看看耗时在哪里么,网络前传应该不需要这么多时间

bubbliiiing commented 1 year ago

image 这个是单3060的速度0 0.

charming2992 commented 1 year ago

image 这个是单3060的速度0 0.

up这个的fps是26.9,我的卡是3090的fps是30。所以是正常的吗?

bubbliiiing commented 1 year ago

不正常啊 3090是3060能力的两三倍呢

bubbliiiing commented 1 year ago

你可以进get fps看看哪里耗时?

charming2992 commented 1 year ago

image 因为报错我改动过一个地方,如上图。所以是这个原因吗?

bubbliiiing commented 1 year ago

有影响,fuse后快一些,合并了bn和conv

2314453446 commented 2 months ago

image 这个是单3060的速度0 0.

up这个的fps是26.9,我的卡是3090的fps是30。所以是正常的吗?

所以有发现问题在哪儿吗,我4090跑出来fps也才三十几