YOLOv4-pytorch icon indicating copy to clipboard operation
YOLOv4-pytorch copied to clipboard

训练精度很高,但FPS比较低

Open whyisma opened this issue 3 years ago • 21 comments

我的训练集是VOC trainval 2012 +VOC trainval 2007,测试集是VOC test 2007,训练采用多尺度训练,mAP可以达到84.1,但FPS比较低,我想问下是什么原因导致的呢? 以下是测试的精度图和推理速度图: mAP QQ图片20200827142503

whyisma avatar Aug 27 '20 06:08 whyisma

额,我觉得你应该把你的设备信息贴出来,不同的设备肯定推理性能不同

zhangyahui520 avatar Aug 31 '20 01:08 zhangyahui520

额,我觉得你应该把你的设备信息贴出来,不同的设备肯定推理性能不同

我用的实验室电脑跑的,CPU I7 9700KF,GPU RTX 2070, 内存16个G。

whyisma avatar Sep 02 '20 08:09 whyisma

我测试的时候,需要十秒才能完成对一张图片的检测。。。

HuanSuper avatar Oct 07 '20 12:10 HuanSuper

我测试的时候,需要十秒才能完成对一张图片的检测。。。

不会吧。。。我检测一张图只用了100毫秒左右

whyisma avatar Oct 07 '20 13:10 whyisma

单张2080ti,下面是我的部分命令行输出 0%| | 0/2495 [00:00<?, ?it/s] 0%| | 1/2495 [00:14<9:47:12, 14.13s/it] 0%| | 2/2495 [00:29<10:05:00, 14.56s/it] 0%| | 3/2495 [00:41<9:33:39, 13.81s/it] 0%| | 4/2495 [00:54<9:15:16, 13.37s/it] 0%| | 5/2495 [01:04<8:40:33, 12.54s/it] 0%| | 6/2495 [01:14<8:11:05, 11.84s/it] 0%| | 7/2495 [01:26<8:04:18, 11.68s/it] 0%| | 8/2495 [01:42<9:07:25, 13.21s/it] 0%| | 9/2495 [01:54<8:51:40, 12.83s/it]

HuanSuper avatar Oct 07 '20 13:10 HuanSuper

单张2080ti,下面是我的部分命令行输出 0%| | 0/2495 [00:00<?, ?it/s] 0%| | 1/2495 [00:14<9:47:12, 14.13s/it] 0%| | 2/2495 [00:29<10:05:00, 14.56s/it] 0%| | 3/2495 [00:41<9:33:39, 13.81s/it] 0%| | 4/2495 [00:54<9:15:16, 13.37s/it] 0%| | 5/2495 [01:04<8:40:33, 12.54s/it] 0%| | 6/2495 [01:14<8:11:05, 11.84s/it] 0%| | 7/2495 [01:26<8:04:18, 11.68s/it] 0%| | 8/2495 [01:42<9:07:25, 13.21s/it] 0%| | 9/2495 [01:54<8:51:40, 12.83s/it]

你这个确实有点慢。。。为啥呢。你训练了几个epoch?

whyisma avatar Oct 07 '20 13:10 whyisma

我用的是作者提供的pretrained权重

HuanSuper avatar Oct 07 '20 13:10 HuanSuper

很奇怪,不知道时间都用到哪去了?

HuanSuper avatar Oct 07 '20 13:10 HuanSuper

我用的是作者提供的pretrained权重

直接用预训练模型来TEST?我没试过,我训练完了来测试还是挺正常的

whyisma avatar Oct 07 '20 13:10 whyisma

请问你训练了多少个epoch,bs是多少,训练了多久噢。感觉这个训练时间好长。。

czzbb avatar Oct 15 '20 12:10 czzbb

请问你训练了多少个epoch,bs是多少,训练了多久噢。感觉这个训练时间好长。。

我训练了300个回合。我自己写了一个多卡的train.py,使用多卡训练,只花了一天时间。

whyisma avatar Oct 30 '20 02:10 whyisma

单张2080ti,下面是我的部分命令行输出 0%| | 0/2495 [00:00<?, ?it/s] 0%| | 1/2495 [00:14<9:47:12, 14.13s/it] 0%| | 2/2495 [00:29<10:05:00, 14.56s/it] 0%| | 3/2495 [00:41<9:33:39, 13.81s/it] 0%| | 4/2495 [00:54<9:15:16, 13.37s/it] 0%| | 5/2495 [01:04<8:40:33, 12.54s/it] 0%| | 6/2495 [01:14<8:11:05, 11.84s/it] 0%| | 7/2495 [01:26<8:04:18, 11.68s/it] 0%| | 8/2495 [01:42<9:07:25, 13.21s/it] 0%| | 9/2495 [01:54<8:51:40, 12.83s/it]

请问你解决这个问题了吗?我也差不多花了这么多时间

zhanghongsir avatar Nov 03 '20 02:11 zhanghongsir

请问你训练了多少个epoch,bs是多少,训练了多久噢。感觉这个训练时间好长。。

我训练了300个回合。我自己写了一个多卡的train.py,使用多卡训练,只花了一天时间。

为啥eval的时候这么慢啊,想不明白

zhanghongsir avatar Nov 03 '20 08:11 zhanghongsir

我的训练集是VOC trainval 2012 +VOC trainval 2007,测试集是VOC test 2007,训练采用多尺度训练,mAP可以达到84.1,但FPS比较低,我想问下是什么原因导致的呢? 以下是测试的精度图和推理速度图: mAP QQ图片20200827142503

你好,想问一下权重文件你用的是哪个啊?yolov4.weights,还是mobilenetv3.pth还是mobilenetv2.pth啊?我用的yolov4.weights,最后的test mAP只有0.589。可以说一下你具体的操作过程吗?

gll-sketch avatar Nov 08 '20 07:11 gll-sketch

请问你训练了多少个epoch,bs是多少,训练了多久噢。感觉这个训练时间好长。。

我训练了300个回合。我自己写了一个多卡的train.py,使用多卡训练,只花了一天时间。

你好,使用多卡的训练文件你是怎么写的啊?可以参考一下吗?

gll-sketch avatar Nov 08 '20 07:11 gll-sketch

我测试的时候,需要十秒才能完成对一张图片的检测。。。

不会吧。。。我检测一张图只用了100毫秒左右

是前向推理速度为100ms还是整个过程用100ms啊?

moxiaoguai1993 avatar Nov 11 '20 11:11 moxiaoguai1993

@whyisma 可以贴一下你的config配置文件吗,想看一下超参如何设置

Imagery007 avatar Nov 19 '20 04:11 Imagery007

请问你训练了多少个纪元,bs是多少,训练了多久噢.感觉这个训练时间好长.

我训练了300个回合.我自己写了一个多卡的Tra.py,使用多卡训练,只花了一天时间。

你好 我想问一下多卡训练代码是怎么弄的

cymdhx avatar Mar 19 '21 11:03 cymdhx

请问你的0.841或者作者的0.851是在cspdarknet上面的结果还是mobilenetv2上面的结果?

Anleeno-Xu avatar May 19 '21 05:05 Anleeno-Xu

我的训练集是VOC trainval 2012 +VOC trainval 2007,测试集是VOC test 2007,训练采用多尺度训练,mAP可以达到84.1,但FPS比较低,我想问下是什么原因导致的呢? 以下是测试的精度图和推理速度图: mAP QQ图片20200827142503

这个图怎么得到的

sjfei avatar May 22 '21 02:05 sjfei

@whyisma can yo please share the pre-trained weights of the author on the google drive ??

abhigoku10 avatar Nov 15 '21 09:11 abhigoku10