nabil_oh/CnD_Classification
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed a typo

Browse Source
This commit is contained in:
Nabil Ould Hamou 2025-01-31 23:23:10 +01:00
parent 0b3a3b79d8
commit 9846ea8c10
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
predict.py
View File

@ -10,7 +10,7 @@ if torch.cuda.is_available():
elif torch.mps.is_available(): elif torch.mps.is_available():
DEVICE = torch.device("mps") DEVICE = torch.device("mps")
else: else:
DEIVCE = torch.device("cpu") DEVICE = torch.device("cpu")
def predict_image(image_path, model): def predict_image(image_path, model):
img = cv2.imread(image_path) img = cv2.imread(image_path)

2
train.py
View File

@ -15,7 +15,7 @@ if torch.cuda.is_available():
elif torch.mps.is_available(): elif torch.mps.is_available():
DEVICE = torch.device("mps") DEVICE = torch.device("mps")
else: else:
DEIVCE = torch.device("cpu") DEVICE = torch.device("cpu")
def load_images_from_folder(folder, label): def load_images_from_folder(folder, label):
data = [] data = []