【问题标题】:Pre-trained model to recognize residential floor plans?识别住宅平面图的预训练模型?
【发布时间】:2021-07-01 08:04:42
【问题描述】:

有没有一种模型可以在给定图像的情况下识别该图像是否是住宅物业的平面图?或者我可以在其中应用迁移学习的预训练通用模型?

这是一个示例image。应该很快就能训练出来,但想检查一下是否有普遍接受的模型。

【问题讨论】:

  • 图片链接失效。
  • @OliverMason 更新了!

标签: dataset artificial-intelligence tensorflow-datasets image-recognition pre-trained-model


【解决方案1】:

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

这是一个来自网络的例子,更多例子你可以搜索“fastai 迁移学习”:https://towardsdatascience.com/transfer-learning-using-the-fastai-library-d686b238213e

最难的部分是获取数据集。 Here 是一个例子,但我真诚地建议您完成免费 fastai course 的第 1 课和第 2 课。它将为您提供更好的概览。

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

【讨论】:

  • 很好的回复,谢谢!我会检查这些资源。应该是一个有趣的快速项目。
猜你喜欢
  • 2019-12-07
  • 2018-11-20
  • 2019-10-30
  • 2017-08-19
  • 1970-01-01
  • 2016-04-03
  • 1970-01-01
  • 2023-04-02
  • 2021-02-19
相关资源
最近更新 更多