526互联

ImageFolder

pytorch 的 torchvision.datasets.ImageFolder 来自定义数据集

import torchvision class ClassificationDataset(torchvision.datasets.ImageFolder): """ YOLOv5 Classification Dataset. Arguments root: Dataset path """ ......
ImageFolder torchvision datasets pytorch 数据
共1篇  :1/1页 首页上一页1下一页尾页