Use imageio instead of matplotlib to generate image.
Some checks failed
continuous-integration/drone/push Build is failing

This commit is contained in:
Mikaël Capelle
2024-12-15 10:13:42 +01:00
parent b15131bf1e
commit fcd4b47951
3 changed files with 31 additions and 352 deletions

View File

@@ -1,5 +1,6 @@
from abc import abstractmethod
from logging import Logger
from pathlib import Path
from typing import Any, Final, Iterable, Iterator, Protocol, Sequence, TypeVar, overload
from numpy.typing import NDArray
@@ -22,10 +23,10 @@ class FileHandler:
def image(self, filename: str, image: NDArray[Any]):
from io import BytesIO
import matplotlib.pyplot as plt
import imageio.v3 as iio
io = BytesIO()
plt.imsave(io, image) # type: ignore
iio.imwrite(io, image, extension=Path(filename).suffix) # type: ignore
io.seek(0)
self.create(filename, io.read(), False)