Но у меня возникли проблемы, когда я решил создать вторую модель с еще десятью изображениями. .
Мои первый и второй коды были похожи, но названия сохраненных моделей были разными.
В качестве IDE я использую Spyder 3.11
Код: Выделить всё
import torch
from PIL import Image
import os
# Step 1: Load the YOLOv5 model
model = torch.hub.load('ultralytics/yolov5', 'yolov5s')
# Step 2: Prepare the training data
dataset_path = 'C:/Users/Stk/Desktop'
image_files = ['eye_1.png', 'eye_2.png', 'eye_3.png', 'eye_4.png', 'eye_5.png']
images = [Image.open(os.path.join(dataset_path, f)) for f in image_files]
# Step 3: Fit the model on the training data
results = model(images)
# Step 4: Inspect the results
display(results.pandas().xyxy[0])
# Step 5: Save the model for use with the camera
model.eval()
save_path = os.path.join(dataset_path, 'yolov5s.pt')
torch.save(model.state_dict(), save_path)
Код: Выделить всё
Exception: 'model'. Cache may be out of date, try `force_reload=True` or see https://docs.ultralytics.com/yolov5/tutorials/pytorch_hub_model_loading for help.
Я потратил около семи дней, и в настоящее время я как сумасшедший.
Подробнее здесь: https://stackoverflow.com/questions/786 ... date-error