PyTorch制作图片数据集
PyTorch制作图片数据集
os.listdir()函数
>>> path = './data/training_set/training_set/cats'
>>> os.listdir(path)
['cat.1.jpg', 'cat.10.jpg', 'cat.100.jpg', 'cat.1000.jpg', 'cat.1001.jpg', 'cat.1002.jpg', 'cat.1003.jpg', 'cat.1004.jpg', 'cat.1005.jpg
原创
2021-03-13 22:29:26 ·
1478 阅读 ·
0 评论