一 把数据集处理好之后,直接端到端训练。结果报这个错误,网上搜寻后最终发现正确的答案有:
batch_size不能整除,但设置为整除后还是报错。
二 再次狂看博客,最终发现是数据集图片的尺寸不一致导致的,即width和height不全是同一个尺寸。因此写程序处理尺寸后得以正常运行
三 附上python对目录操作的常见指令
- 删除目录 os.remove(path)
- 新建目录 os.mkdir(path)
- 移动文件 shutil.copy("old_path", "new_path")支持右斜杠\
- 重命名文件 os.rename(old,new)
- 遍历目录 fileList=os.listdir(path) 而后遍历fileList即可