【发布时间】:2017-09-28 22:43:36
【问题描述】:
我有这两组图像。每张图片都附有一个百分比。我有一个使用返回二进制结果(0 或 1)的图像训练的模型。我想返回机器对 0 或 1 的确定程度的百分比。
举个例子,假设它们是猫图像。我也有关于“猫”有多大的百分比(100% 非常大,0% 非常小)。我希望模型告诉我“这是一只 78% 大的猫”。
我当前的数据集基于两个文件夹(大猫和小猫)。第一个数字是图像的编号(例如 1 到 1000)。第二个数字是猫有多大的百分比:
../big_cats
cat1_78.png
cat2_97.png
cat3_67.png
...
../small_cats
cat1_12.png
cat2_9.png
cat3_24.png
...
所以现在我不使用百分比,我得到一个 0 或 1(大或小),但我的目标是模型使用它来告诉我“这是一只 67% 的大猫” ,其中 50% 是中型猫,100% 是最大的猫,0% 是最小的猫。
我尝试过使用softmax 和categorical_cassentropy,但这只是说明猫是大还是小的二元决策的估计准确度(如果我理解正确的话)。
我的问题是:如何准备数据集以获得该结果?如何创建数据训练和验证以读取图像名称中的这个百分比?有什么我可以学习的教程吗?我需要使用Keras + Tensorflow。
【问题讨论】:
标签: machine-learning tensorflow keras