#python #torch
#python #факел
Вопрос:
Я использую torch.utils.data.DataLoader для поиска имени файла в наборе данных в итеративном стиле. При вызове DataLoader.dataset.samples[i] в итерационном цикле я получаю имя файла и целое число. Пример:
('car.JPEG', 0)
Что означает второй параметр?
Ответ №1:
DataLoader Возвращает каждый элемент dataset объекта, который является кортежем, содержащим образец и class_index, к которому он принадлежит. Итак, в вашем случае первый индекс кортежа содержит sample ( car.JEPG ), а второй индекс — class_index, которому принадлежит образец ( 0 в вашем случае). Чтобы узнать больше, ознакомьтесь с документами.