使用TurboJPG环境加速mmdetection训练

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’

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值