mmdet里LoadImageFromFile时选用这个TurboJPG(https://github.com/lilohuang/PyTurboJPEG/),图片解码读取速度可以提高2-3倍,
可以降低我们多人训练使用服务器CPU瓶颈,加速大家训练时间。
172.16.1.98安装步骤:
1.切换到自己的conda环境,如source activate 【环境名】。
2.安装TurboJPEG第三方库,安装命令:pip install -U git+git://github.com/lilohuang/PyTurboJPEG.git
3.找到conda环境下的mmcv第三方库文件io.py,比如我自己的
路径是:/home/szhang/.conda/envs/mmd_zs_3.7/lib/python3.7/site-packages/mmcv/image/io.py
4.修改io.py文件,
大约在文件12行修改如下:
try:
from turbojpeg import TJCS_RGB, TJPF_BGR, TJPF_GRAY, TurboJPEG
jpeg = TurboJPEG()
except ImportError:
TJCS_RGB = TJPF_GRAY = TJPF_BGR = TurboJPEG = None
jpeg = None
大约在文件24行修改如下:
将jpeg = None代码注释掉。
大约在文件33行修改如下:
imread_backend = ‘turbojpeg’