yolov5 model while trying to create a confusion matrix, I encountered an issue. Please help. #5173
Labels
invalid/unrelated
unrelated to this project or invalid type of issues
needs-more-info
More info is needed to complete the issue
code :
import torch
from torch.utils.data import DataLoader
from torchvision import transforms
from torchvision.datasets import ImageFolder
from sklearn.metrics import confusion_matrix, classification_report
import numpy as np
from tqdm import tqdm
from PIL import Image
data_dir = "/content/testdataset"
transform = transforms.Compose([
transforms.Resize((224, 224)),
transforms.ToTensor(),
])
dataset = ImageFolder(data_dir, transform=transform)
dataloader = DataLoader(dataset, batch_size=32, shuffle=False)
print(dataset)
output :
Dataset ImageFolder
Number of datapoints: 1491
Root location: /content/testdataset
StandardTransform
Transform: Compose(
Resize(size=(224, 224), interpolation=bilinear, max_size=None, antialias=warn)
ToTensor()
)
I was fine until now, but I'm encountering issues from this point onward.
code :
model.eval()
all_predictions = []
all_labels = []
for inputs, labels in tqdm(dataloader, desc="Evaluating"):
# 모델 예측 수행
with torch.no_grad():
outputs = model(inputs)
----> 8 for inputs, labels in tqdm(dataloader, desc="Evaluating"):
UnidentifiedImageError: cannot identify image file <_io.BufferedReader name='/content/testdataset/with_mask/1 (371).jpeg'>
The text was updated successfully, but these errors were encountered: