Changes in symbol.py

Trying just bytes now
This commit is contained in:
Christoph Eder committed 2025-12-18 16:18:36 +01:00
1 parent af2abad1ff
commit a4799f2c3f
1 file changed
+3 -5
+3 -5
View File
@@ -11,21 +11,19 @@ import base64
class Symbol:
id: int
label: str
image: np.ndarray
image: bytes
@classmethod
def from_dict(cls, data: dict) -> "Symbol":
if isinstance(data["image"], str):
# Decode base64 to numpy array
image_bytes = base64.b64decode(data["image"])
image_array = np.frombuffer(image_bytes, dtype=np.uint8)
else:
image_array = np.array(data["image"], dtype=np.uint8)
image_bytes = bytes(data["image"])
return cls(
id=data["id"],
label=data["label"],
image=image_array,
image=image_bytes,
)
@classmethod