From eb2cb6139bf63d58e2b541ea5d63eeac57e57bce Mon Sep 17 00:00:00 2001
From: zhangsong <songz.qqmail@qq.com>
Date: Sun, 29 Jun 2025 14:04:25 +0800
Subject: [PATCH] support turbojpeg for mmcv imread

---
 mmcv/image/io.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/mmcv/image/io.py b/mmcv/image/io.py
index e10d443da6..c609543fe1 100644
--- a/mmcv/image/io.py
+++ b/mmcv/image/io.py
@@ -265,6 +265,12 @@ def imfrombytes(content: bytes,
             f'backend: {backend} is not supported. Supported '
             "backends are 'cv2', 'turbojpeg', 'pillow', 'tifffile'")
     if backend == 'turbojpeg':
+        if TurboJPEG is None:
+              raise ImportError('`PyTurboJPEG` is not installed')
+        global jpeg
+        if jpeg is None:
+            jpeg = TurboJPEG()
+
         img = jpeg.decode(  # type: ignore
             content, _jpegflag(flag, channel_order))
         if img.shape[-1] == 1: