卷积神经网络对房屋/公寓的不同房间的图片进行分类,在1839幅图像上具有88.9%的验证准确度。
GitHub链接
https://github.com/ironhide23586/RoomNet
完整演示文稿
https://github.com/ironhide23586/RoomNet/raw/master/documentation/RoomNet%20Presentation.pptx
这是一个自定义神经网络,我将输入图像分类为以下6个类之一(按其类ID的顺序) -
建筑
建筑模块
完整的网络架构
开箱即用的推论
infer.py中的优化推理代码。请参阅main方法中调用classify_im_dir方法的短代码。
训练
转换为推理优化版本
训练环境
以前的方法尝试过
表现图
验证准确性
验证分类F-Score
验证分类精度
验证分类回忆