Я пытаюсь создать классификатор изображений AI на Python, используя руководство на YouTube. Когда я запускаю свою програ ⇐ Python

[Anonymous]

Я пытаюсь создать классификатор изображений AI, используя руководство YouTube для школьного проекта. Вот ссылка:
На этом этапе я еще не закончил, но когда я запускаю свой main.py Я получаю следующую ошибку:

Код: Выделить всё

Traceback (most recent call last):
File "c:\xxx\xx\xx\xx\newai\main.py", line 19, in 
for img_path in os.listdir(os.path.join(dir_, category)):
NotADirectoryError: [WinError 267] The directory name is invalid: 'C:/xx/xx/xx/xx/newai\\Data\\Blue-Squares\\BlueSquare (1).jpg'

Я тоже это понимаю, но не думаю, что лично для проекта это имеет большое значение. (Может быть, и так, я просто предполагаю, что, поскольку это работает в видео, оно все равно должно работать, поскольку видео появилось недавно.) :

Код: Выделить всё

:\XX\XX\XX\XX\Packages\PythonSoftwareFoundation.Python.3.9_qbz5n2kfra8p0\LocalCache\local-packages\Python39\site-packages\torchvision\models\_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead.
warnings.warn(
C:\XX\XX\XX\XX\Packages\PythonSoftwareFoundation.Python.3.9_qbz5n2kfra8p0\LocalCache\local-packages\Python39\site-packages\torchvision\models\_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=ResNet18_Weights.IMAGENET1K_V1`. You can also use `weights=ResNet18_Weights.DEFAULT` to get the most up-to-date weights.
warnings.warn(msg)

На видео, когда он запускает его на этом этапе, он печатает ключи и работает нормально.
Вот мой полный код:

Код: Выделить всё

from img2vec_pytorch import Img2Vec
import os
from PIL import Image

# prepare data

img2vec = Img2Vec()

data_dir = 'C:/XX/XX/XX/XX/newai'
train_dir = os.path.join(data_dir, r'Data', r'Blue-Squares')
val_dir = os.path.join(data_dir, r'Data', r'Red-Triangles')

data = {}

for j, dir_ in enumerate([train_dir, val_dir]):
features = []
labels = []
for category in os.listdir(dir_):
for img_path in os.listdir(os.path.join(dir_, category)):
img_path_ = os.path.join(dir_, category, img_path)
img = Image.open(img_path_)

img_features = img2vec.get_vec(img)

features.append(img_features)
labels.append(category)

data[['training_data', 'validation_data'][j]] = features
data[['training_labels', 'validation_labels'][j]] = labels

print(data.keys())

# train model

# test performance

# save the model

Я пробовал:
скопировать и вставить код YouTubers и использовать именно его, поменять пути, изменить папки, изменить имена изображений, изменить способ их настройки, поискать ошибки в Google и т. д. Я знаю, что изображение не является каталогом, поэтому я понимаю, что просто не понимаю, что изменить. Мы будем очень признательны за любые отзывы.

Подробнее здесь: https://stackoverflow.com/questions/782 ... -guide-whe