猫狗分类
来源:6-10 10monkeys基础模型搭建与训练
Riddle2000
2020-05-26
Hi老师,
最近做了猫狗分类的问题。但是数据集里的每个图片只有一个猫,或者一个狗。
如果图片里同时有一个猫和一个狗,我期望预测结果是“猫和狗”,请问我要学什么能解决这个问题。
另外一个问题,如果我只想让模型识别猫或者狗,给模型输入其他图片的时候,模型输出其他,这种要怎么训练?
谢谢老师
写回答
1回答
-
同时预测猫狗的问题,终极解决办法是学习目标检测把不同的物体给圈出来,然后去做预测。一个取巧的办法,是把图像切分成很多块,输入给模型,看模型会不会在猫和狗这两类上分别能有比较高的概率值。
其他类的这个问题,需要把二分类问题变成三分类问题,还需要加一些没有猫狗的图像参与训练。
112020-06-16
相似问题