-
Notifications
You must be signed in to change notification settings - Fork 49
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
使用GPU运行车牌检查,有警告,运行耗时比CPU #2
Comments
使用命令sudo python test.py / home / zjs / Desktop /车牌测试/ --no_yolo --beam --gpu 这个警告应该如何解决呢?并且在使用GPU的情况下,为什么耗时对比CPU更大了呢?我应该如何修改? |
这个并不是警告,那是mxnet自动选择卷积算法的提示,这是一个比较费时的操作,如果需要屏蔽按提示设置环境变量即可 |
另外,因为模型规模本身不算大,单张图片使用GPU并不能有很好的加速效果。 |
对于车牌检测的图片大小有什么要求吗?比如640×480 |
我修改了环境变量,添加了export MXNET_CUDNN_AUTOTUNE_DEFAULT=0 |
对于车牌检测的图片大小有什么要求吗?比如 640×480 |
我这边设置好环境变量过后就可以了,耗时就会大幅度降低。或者你试试这样: MXNET_CUDNN_AUTOTUNE_DEFAULT=0 python3 test.py --beam --gpu /path/to/image 模型是全卷积网络,理论上对图片大小没有要求。只是过大的图片对提高准确率并没有太大的帮助,所以你自己根据情况找一个效率和准确率的平衡点就好。 |
好的,我试一下,感谢您的回答 |
No description provided.
The text was updated successfully, but these errors were encountered: