我正在按照Tensorflow的教程学习机器学习模型的训练。我已经从我的电脑上传了自己的数据集到Google Colab中的一个名为“sample_arrow”的文件夹,并指定了路径:
image_path = 'sample_arrow'
这个文件夹里有图像,大小不为0。但是在执行以下代码时,我得到了一个错误:
data = DataLoader.from_folder(image_path) train_data, test_data = data.split(0.9)ValueError: Image size is zero
这里出了什么问题?可能是文件夹路径指定不正确吗?我对这个话题完全是新手,对Python也不熟悉(但有Java技能),希望能得到详细的回答。
回答:
最后,我找到了解决方案。缺少导入os模块和正确定义路径:
import osroot_path = "/content/" image_path = os.path.join(os.path.dirname(root_path), 'sample_arrow')