lamonov.pavel/PaddleOCR
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Переделал метод для создания заглушки.
Some checks are pending
Build/Publish Develop Docs / deploy (push) Waiting to run
Details

Browse Source
This commit is contained in:
Ламонов Павел
2025-07-02 14:40:44 +03:00
parent 304db2ab27
commit 766386507c
1 changed files with 10 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
ppocr/data/simple_dataset.py
View File

@@ -118,15 +118,19 @@ class SimpleDataSet(Dataset):
def create_dummy_sample(self): def create_dummy_sample(self):
"""Создает пустой образец для предотвращения сбоев""" """Создает пустой образец для предотвращения сбоев"""
dummy_img = np.zeros((32, 320, 3), dtype=np.uint8) dummy_img = np.ones((32, 320, 3), dtype=np.uint8) * 255
dummy_img = cv2.imencode('.jpg', dummy_img)[1].tobytes() 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 { return {
"image": dummy_img, "image": img_bytes,
"label": "", "label": "DUMMY",
"img_path": "dummy.jpg", "img_path": "/kaggle/input/dummy.jpg",
"ext_data": self.get_ext_data(), "ext_data": self.get_ext_data(),
"filename": "dummy.jpg" "filename": "/kaggle/input/dummy.jpg"
} }
def __getitem__(self, idx): def __getitem__(self, idx):