如果我正在训练一个对情况有两个可接受答案的模型怎么办?

时间:2019-10-07 09:00:51

标签: tensorflow

我想训练一个模型在图像中找到一个蓝点。但是大约有30%的时间有两个蓝点,在这种情况下,我很乐意找到一个蓝点。

我不想批评机器选择“错误点”,因为这并不重要。我只需要找到两个点即可。

我在每幅图像中都有一个蓝点坐标列表,并带有手工标记。

我不确定如何训练它。

我正在将TensorFlow与keras一起使用,当我忽略具有两个的图像时,我可以共享我的代码以找到一个蓝点,但是如果没有很大的数据集,它并不会特别有用。

1 个答案:

答案 0 :(得分:0)

我不确定这是否是一个好答案,但这就是我的目标。而不是给出“蓝点”的“ x,y坐标”。相反,我生成了一个新的图像,该图像的蓝色像素位于输入中的蓝色点所在的位置,然后编写了一个愚蠢的算法,对图像中的每个像素进行暴力破解以找到蓝色像素,并将其放入常规列表中。