你们这个代码不支持gpu？

[MtYCNN]

你们的代码为什么运行起来这么慢，生成一张海报要30多秒。看了看基本没用上GPU啊

[chuhaojin]

支持gpu的，或许看看是否pytorch版本的问题？在你的环境下运行torch.cuda.is_available()看看是否使用了GPU？

[MtYCNN]

测试了GPU可用。pytorch也没问题，也是GPU版本，运行别的程序都没问题。。。但是一跑您这个就很慢
我是用的你们提供的run.sh脚本，我看了一下计算NMS那里特别慢，假如有五个语句，每个语句进行一次NMS的时间是6、7秒，这就要30多秒了。。。。。
但是我测试你们的quick_start.py程序的时候就比较快
请问这是什么原因呢

[chuhaojin]

quick_start.py也是基于这个repo代码搭建的，没什么优化。。或许你看看是不是图像的尺寸太大了，如果图像很大的话就会很慢。也可以再检查一下版本问题和环境配置。。

[MtYCNN]

quick_start.py也是基于这个repo代码搭建的，没什么优化。。或许你看看是不是图像的尺寸太大了，如果图像很大的话就会很慢。也可以再检查一下版本问题和环境配置。。

你们这个不是默认会resize到300*400分辨率吗，而且我看了一下，感觉和quick_start.py里面下载的图片的尺寸差不多啊。。。。。