From 766386507c6ee5db0de550b8ff3ee31a9d0b6100 Mon Sep 17 00:00:00 2001 From: "pavel.lamonov" Date: Wed, 2 Jul 2025 14:40:44 +0300 Subject: [PATCH] =?UTF-8?q?=D0=9F=D0=B5=D1=80=D0=B5=D0=B4=D0=B5=D0=BB?= =?UTF-8?q?=D0=B0=D0=BB=20=D0=BC=D0=B5=D1=82=D0=BE=D0=B4=20=D0=B4=D0=BB?= =?UTF-8?q?=D1=8F=20=D1=81=D0=BE=D0=B7=D0=B4=D0=B0=D0=BD=D0=B8=D1=8F=20?= =?UTF-8?q?=D0=B7=D0=B0=D0=B3=D0=BB=D1=83=D1=88=D0=BA=D0=B8.?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- ppocr/data/simple_dataset.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/ppocr/data/simple_dataset.py b/ppocr/data/simple_dataset.py index e470d46..bd1d4bd 100644 --- a/ppocr/data/simple_dataset.py +++ b/ppocr/data/simple_dataset.py @@ -118,15 +118,19 @@ class SimpleDataSet(Dataset): def create_dummy_sample(self): """Создает пустой образец для предотвращения сбоев""" - dummy_img = np.zeros((32, 320, 3), dtype=np.uint8) - dummy_img = cv2.imencode('.jpg', dummy_img)[1].tobytes() + dummy_img = np.ones((32, 320, 3), dtype=np.uint8) * 255 + success, encoded_img = cv2.imencode('.jpg', dummy_img) + if success: + img_bytes = encoded_img.tobytes() + else: + img_bytes = b"\xff\xd8\xff\xe0\x00\x10JFIF\x00\x01\x01\x00\x00\x01\x00\x01\x00\x00\xff\xdb\x00C\x00\x03\x02\x02\x02\x02\x02\x03\x02\x02\x02\x03\x03\x03\x03\x04\x06\x04\x04\x04\x04\x04\x08\x06\x06\x05\x06\t\x08\n\n\t\x08\t\t\n\x0c\x0f\x0c\n\x0b\x0e\x0b\t\t\r\x11\r\x0e\x0f\x10\x10\x11\x10\n\x0c\x12\x13\x12\x10\x13\x0f\x10\x10\x10\xff\xc0\x00\x0b\x08\x00 \x00\x01\x01\x11\x00\xff\xc4\x00\x14\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xff\xda\x00\x08\x01\x01\x00\x00?\x00\xff\xd9" return { - "image": dummy_img, - "label": "", - "img_path": "dummy.jpg", + "image": img_bytes, + "label": "DUMMY", + "img_path": "/kaggle/input/dummy.jpg", "ext_data": self.get_ext_data(), - "filename": "dummy.jpg" + "filename": "/kaggle/input/dummy.jpg" } def __getitem__(self, idx):