train_dir = os.path.join(base_dir, 'train')
if not os.path.exists(train_dir):
    os.mkdir(train_dir)

写上这条语句能够再次运行代码,再次运行代码不再报错。

欢迎关注我的CSDN博客心系五道口,有问题请私信2395856915@qq.com