Some further speedups.
This commit is contained in:
@ -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
|
||||
|
||||
|
||||
|
||||
Reference in New Issue
Block a user