用于识别住宅平面图的预训练模型？

如何解决用于识别住宅平面图的预训练模型？

是否有模型可以根据图像识别该图像是否是住宅物业的平面图？或者一个经过预训练的通用模型，我可以在其中应用迁移学习？

这是一个示例 image。应该很快训练一个，但想检查是否有一个普遍接受的模型。

解决方法

fastai 具有用于 imagenet 的预训练模型，您可以对其进行重用和迁移学习。您可能可以使用预先训练的 resnet 网络，然后在您想出的数据集上重新训练它。

这是网络上的一个例子，但您可以搜索“fastai transfer learning”以获取更多示例：https://towardsdatascience.com/transfer-learning-using-the-fastai-library-d686b238213e

最难的部分是您获得数据集。 Here 是一个示例，但老实说，我建议您学习免费的 fastai course 的第 1 课和第 2 课。它将为您提供更好的概览。

TL;DR - 如果您可以在谷歌上搜索“住宅平面图”或其他内容，您就可以创建一个数据集。困难的部分是选择要包含在您的数据集中的非楼层计划。可能你需要很多随机的东西，还有很多看起来接近住宅平面图的东西，但不是，所以它可以很好地区分平面图和迷宫、弹球布局和电子表格等。