In: from torchvision.datasets import ImageFolder dataset = ImageFolder("D:/pycharm/PyTorch/dogcatdataset/dogcat/") print(dataset.class_to_idx) Out: ...
確定! 回上一頁