Some further speedups.

This commit is contained in:
Daniel
2024-12-02 17:00:04 +01:00
parent c8c48e4bf2
commit dc44a71b2c
4 changed files with 202 additions and 179 deletions

View File

@ -220,7 +220,7 @@ def update_sample(sample, new_dir=""):
def load_image(path: str):
image = cv2.imread(path, 3)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = np.array(image, dtype=np.float32)
image = np.asarray(image, dtype=np.uint8)
return image