颜色分类leetcode FCNN example 这是一个完全卷积的神经网络练习,用于从航拍图像中检测房屋
颜色分类leetcode FCNN示例这是一份工作申请的练习(我不会透露公司名称)。目标是过度拟合给定的单个图像,一个大的航拍图像以及地面实况,以便我们可以检测房屋。给出了网络架构,所以没有灵活性。这个repo包含PyTorch代码和其他材料来描述我对给定问题的方法。请注意,自定义交叉熵函数实际上是不必要的,将在下一个版本中删除。回购结构images文件夹包含示例图像及其基本事实plots文件夹包含训练模型的损失和得分图predicted_images文件夹包含用训练模型的结果预测标记的图像weights文件夹包含模型权重,其中一个模型通常可能有多个检查点FCNN.py是练习中定义的完全卷积神经网络模型。 Upscaling是常数,默认情况下使用最近邻插值。 FCNN2.py几乎与定义的FCNN.py模型相同,但具有可学习的放大层,即转置卷积,遵循[1]。 train.py是使用给定参数为给定时期训练网络的脚本。它也是:保存包含每次迭代训练损失的图保存包含每个时期的模型分数(详见下文)的图保存得分最高的模型权重集以及最后一组权重。 predi
下载地址
用户评论