飞桨常规赛:遥感影像地块分割 - 4月第8名方案

P粉084495128
发布: 2025-08-01 10:24:17
原创
1024人浏览过
该项目为遥感影像地块分割四分类任务,获常规赛4月第八名。基于PaddleSeg,用OCRNet模型(HRNet为backbone),按0.97:0.03划分训练集与验证集。训练配置含batch_size=64等,用联合损失改善类别不平衡,后修改代码测试并保存结果,压缩输出。

☞☞☞AI 智能聊天, 问答助手, AI 智能搜索, 免费无限量使用 DeepSeek R1 模型☜☜☜

飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网

常规赛:遥感影像地块分割4月第八名

赛题简介

赛题任务 本赛题旨在对遥感影像进行像素级内容解析,并对遥感影像中感兴趣的类别进行提取和分类,以衡量遥感影像地块分割模型在多个类别(如建筑、道路、林地等)上的效果。

数据说明 本赛题提供了多个地区已脱敏的遥感影像数据,各参赛选手可以基于这些数据构建自己的地块分割模型。

训练数据集 样例图片及其标注如下图所示:

飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网 飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网 飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网 飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网        

训练数据集文件名称:train_and_label.zip 包含2个子文件,分别为:训练数据集(原始图片)文件、训练数据集(标注图片)文件,详细介绍如下:

训练数据集(原始图片)文件名称:img_train 包含66,653张分辨率为2m/pixel,尺寸为256 * 256的JPG图片,每张图片的名称形如T000123.jpg。

训练数据集(标注图片)文件名称:lab_train 包含66,653张分辨率为2m/pixel,尺寸为256 * 256的PNG图片,每张图片的名称形如T000123.png。 备注: 全部PNG图片共包括4种分类,像素值分别为0、1、2、3。此外,像素值255为未标注区域,表示对应区域的所属类别并不确定,在评测中也不会考虑这部分区域。

测试数据集 测试数据集文件名称:img_test.zip,详细介绍如下: 包含4,609张分辨率为2m/pixel,尺寸为256 * 256的JPG图片,文件名称形如123.jpg。

本次赛题是四分类的航空影像语义分割任务,需要考虑各类别像素不均衡的问题。

本项目基于PaddleSeg进行训练与推理

首先下载paddleseg,并安装所需的环境

In [ ]
!git clone https://gitee.com/PaddlePaddle/PaddleSeg
!pip install paddleseg
登录后复制
   

解压数据集并对数据集进行训练集和验证集的划分

In [ ]
!unzip data/data77571/img_test.zip
登录后复制
       
  inflating: img_testA/5263.jpg      
  inflating: img_testA/3612.jpg      
  inflating: img_testA/2524.jpg      
  inflating: img_testA/4155.jpg      
  inflating: img_testA/9317.jpg      
  inflating: img_testA/6024.jpg      
  inflating: img_testA/1993.jpg      
  inflating: img_testA/9303.jpg      
  inflating: img_testA/6030.jpg      
  inflating: img_testA/1987.jpg      
  inflating: img_testA/3148.jpg      
  inflating: img_testA/2530.jpg      
  inflating: img_testA/6756.jpg      
  inflating: img_testA/432.jpg       
  inflating: img_testA/354.jpg       
  inflating: img_testA/1818.jpg      
  inflating: img_testA/8182.jpg      
  inflating: img_testA/340.jpg       
  inflating: img_testA/4974.jpg      
  inflating: img_testA/6805.jpg      
  inflating: img_testA/6193.jpg      
  inflating: img_testA/1824.jpg      
  inflating: img_testA/4784.jpg      
  inflating: img_testA/4948.jpg      
  inflating: img_testA/2487.jpg      
  inflating: img_testA/5303.jpg      
  inflating: img_testA/8627.jpg      
  inflating: img_testA/397.jpg       
  inflating: img_testA/5465.jpg      
  inflating: img_testA/5471.jpg      
  inflating: img_testA/7266.jpg      
  inflating: img_testA/8155.jpg      
  inflating: img_testA/8633.jpg      
  inflating: img_testA/4009.jpg      
  inflating: img_testA/9505.jpg      
  inflating: img_testA/7528.jpg      
  inflating: img_testA/4021.jpg      
  inflating: img_testA/2450.jpg      
  inflating: img_testA/5459.jpg      
  inflating: img_testA/3028.jpg      
  inflating: img_testA/4747.jpg      
  inflating: img_testA/9263.jpg      
  inflating: img_testA/6150.jpg      
  inflating: img_testA/9277.jpg      
  inflating: img_testA/6144.jpg      
  inflating: img_testA/2322.jpg      
  inflating: img_testA/4753.jpg      
  inflating: img_testA/4035.jpg      
  inflating: img_testA/2444.jpg      
  inflating: img_testA/3982.jpg      
  inflating: img_testA/6622.jpg      
  inflating: img_testA/156.jpg       
  inflating: img_testA/630.jpg       
  inflating: img_testA/9920.jpg      
  inflating: img_testA/9934.jpg      
  inflating: img_testA/5868.jpg      
  inflating: img_testA/8394.jpg      
  inflating: img_testA/4586.jpg      
  inflating: img_testA/1398.jpg      
  inflating: img_testA/7925.jpg      
  inflating: img_testA/80.jpg        
  inflating: img_testA/9908.jpg      
  inflating: img_testA/4592.jpg      
  inflating: img_testA/5854.jpg      
  inflating: img_testA/4579.jpg      
  inflating: img_testA/3216.jpg      
  inflating: img_testA/5667.jpg      
  inflating: img_testA/7070.jpg      
  inflating: img_testA/8425.jpg      
  inflating: img_testA/7716.jpg      
  inflating: img_testA/5101.jpg      
  inflating: img_testA/3570.jpg      
  inflating: img_testA/3564.jpg      
  inflating: img_testA/7702.jpg      
  inflating: img_testA/1373.jpg      
  inflating: img_testA/9049.jpg      
  inflating: img_testA/181.jpg       
  inflating: img_testA/7064.jpg      
  inflating: img_testA/6352.jpg      
  inflating: img_testA/5883.jpg      
  inflating: img_testA/5.jpg         
  inflating: img_testA/4223.jpg      
  inflating: img_testA/817.jpg       
  inflating: img_testA/8419.jpg      
  inflating: img_testA/9713.jpg      
  inflating: img_testA/4237.jpg      
  inflating: img_testA/4551.jpg      
  inflating: img_testA/5897.jpg      
  inflating: img_testA/9075.jpg      
  inflating: img_testA/1359.jpg      
  inflating: img_testA/7728.jpg      
  inflating: img_testA/815.jpg       
  inflating: img_testA/2888.jpg      
  inflating: img_testA/2650.jpg      
  inflating: img_testA/5659.jpg      
  inflating: img_testA/4547.jpg      
  inflating: img_testA/8369.jpg      
  inflating: img_testA/9077.jpg      
  inflating: img_testA/2122.jpg      
  inflating: img_testA/5895.jpg      
  inflating: img_testA/6422.jpg      
  inflating: img_testA/9711.jpg      
  inflating: img_testA/5103.jpg      
  inflating: img_testA/69.jpg        
  inflating: img_testA/8427.jpg      
  inflating: img_testA/829.jpg       
  inflating: img_testA/7072.jpg      
  inflating: img_testA/1403.jpg      
  inflating: img_testA/5665.jpg      
  inflating: img_testA/5671.jpg      
  inflating: img_testA/8355.jpg      
  inflating: img_testA/1417.jpg      
  inflating: img_testA/6378.jpg      
  inflating: img_testA/1371.jpg      
  inflating: img_testA/8433.jpg      
  inflating: img_testA/4209.jpg      
  inflating: img_testA/2678.jpg      
  inflating: img_testA/2693.jpg      
  inflating: img_testA/96.jpg        
  inflating: img_testA/7933.jpg      
  inflating: img_testA/168.jpg       
  inflating: img_testA/6393.jpg      
  inflating: img_testA/5842.jpg      
  inflating: img_testA/6387.jpg      
  inflating: img_testA/7099.jpg      
  inflating: img_testA/82.jpg        
  inflating: img_testA/7927.jpg      
  inflating: img_testA/2687.jpg      
  inflating: img_testA/2877.jpg      
  inflating: img_testA/154.jpg       
  inflating: img_testA/8382.jpg      
  inflating: img_testA/140.jpg       
  inflating: img_testA/8396.jpg      
  inflating: img_testA/9936.jpg      
  inflating: img_testA/626.jpg       
  inflating: img_testA/9261.jpg      
  inflating: img_testA/4745.jpg      
  inflating: img_testA/2334.jpg      
  inflating: img_testA/3994.jpg      
  inflating: img_testA/2452.jpg      
  inflating: img_testA/4023.jpg      
  inflating: img_testA/8619.jpg      
  inflating: img_testA/9507.jpg      
  inflating: img_testA/6634.jpg      
  inflating: img_testA/6620.jpg      
  inflating: img_testA/5329.jpg      
  inflating: img_testA/3980.jpg      
  inflating: img_testA/3758.jpg      
  inflating: img_testA/4037.jpg      
  inflating: img_testA/4751.jpg      
  inflating: img_testA/4989.jpg      
  inflating: img_testA/9275.jpg      
  inflating: img_testA/6146.jpg      
  inflating: img_testA/1629.jpg      
  inflating: img_testA/4779.jpg      
  inflating: img_testA/7270.jpg      
  inflating: img_testA/8625.jpg      
  inflating: img_testA/1167.jpg      
  inflating: img_testA/5301.jpg      
  inflating: img_testA/3770.jpg      
  inflating: img_testA/1173.jpg      
  inflating: img_testA/1615.jpg      
  inflating: img_testA/7264.jpg      
  inflating: img_testA/8157.jpg      
  inflating: img_testA/1826.jpg      
  inflating: img_testA/8802.jpg      
  inflating: img_testA/1198.jpg      
  inflating: img_testA/2491.jpg      
  inflating: img_testA/3943.jpg      
  inflating: img_testA/418.jpg       
  inflating: img_testA/8816.jpg      
  inflating: img_testA/4792.jpg      
  inflating: img_testA/4962.jpg      
  inflating: img_testA/430.jpg       
  inflating: img_testA/6807.jpg      
  inflating: img_testA/1749.jpg      
  inflating: img_testA/6026.jpg      
  inflating: img_testA/4157.jpg      
  inflating: img_testA/3638.jpg      
  inflating: img_testA/2526.jpg      
  inflating: img_testA/5249.jpg      
  inflating: img_testA/6998.jpg      
  inflating: img_testA/9467.jpg      
  inflating: img_testA/6754.jpg      
  inflating: img_testA/2254.jpg      
  inflating: img_testA/4625.jpg      
  inflating: img_testA/9301.jpg      
  inflating: img_testA/6032.jpg      
  inflating: img_testA/5513.jpg      
  inflating: img_testA/3162.jpg      
  inflating: img_testA/8037.jpg      
  inflating: img_testA/8989.jpg      
  inflating: img_testA/1013.jpg      
  inflating: img_testA/8751.jpg      
  inflating: img_testA/6768.jpg      
  inflating: img_testA/8023.jpg      
  inflating: img_testA/1761.jpg      
  inflating: img_testA/2268.jpg      
  inflating: img_testA/578.jpg       
  inflating: img_testA/8976.jpg      
  inflating: img_testA/4194.jpg      
  inflating: img_testA/3837.jpg      
  inflating: img_testA/8962.jpg      
  inflating: img_testA/1946.jpg      
  inflating: img_testA/3189.jpg      
  inflating: img_testA/2297.jpg      
  inflating: img_testA/6967.jpg      
  inflating: img_testA/544.jpg       
  inflating: img_testA/550.jpg       
  inflating: img_testA/4802.jpg      
  inflating: img_testA/7884.jpg      
  inflating: img_testA/9671.jpg      
  inflating: img_testA/2724.jpg      
  inflating: img_testA/4355.jpg      
  inflating: img_testA/2042.jpg      
  inflating: img_testA/6224.jpg      
  inflating: img_testA/9117.jpg      
  inflating: img_testA/8209.jpg      
  inflating: img_testA/9103.jpg      
  inflating: img_testA/4427.jpg      
  inflating: img_testA/2056.jpg      
  inflating: img_testA/5739.jpg      
  inflating: img_testA/2730.jpg      
  inflating: img_testA/975.jpg       
  inflating: img_testA/3406.jpg      
  inflating: img_testA/8553.jpg      
  inflating: img_testA/7660.jpg      
  inflating: img_testA/9895.jpg      
  inflating: img_testA/6218.jpg      
  inflating: img_testA/8235.jpg      
  inflating: img_testA/3360.jpg      
  inflating: img_testA/5711.jpg      
  inflating: img_testA/1563.jpg      
  inflating: img_testA/7112.jpg      
  inflating: img_testA/949.jpg       
  inflating: img_testA/7674.jpg      
  inflating: img_testA/1205.jpg      
  inflating: img_testA/9659.jpg      
  inflating: img_testA/7847.jpg      
  inflating: img_testA/1588.jpg      
  inflating: img_testA/2081.jpg      
  inflating: img_testA/6595.jpg      
  inflating: img_testA/746.jpg       
  inflating: img_testA/8590.jpg      
  inflating: img_testA/9842.jpg      
  inflating: img_testA/9843.jpg      
  inflating: img_testA/2916.jpg      
  inflating: img_testA/2902.jpg      
  inflating: img_testA/9857.jpg      
  inflating: img_testA/8591.jpg      
  inflating: img_testA/747.jpg       
  inflating: img_testA/5923.jpg      
  inflating: img_testA/6580.jpg      
  inflating: img_testA/7846.jpg      
  inflating: img_testA/5089.jpg      
  inflating: img_testA/4397.jpg      
  inflating: img_testA/1589.jpg      
  inflating: img_testA/7113.jpg      
  inflating: img_testA/8220.jpg      
  inflating: img_testA/3413.jpg      
  inflating: img_testA/5062.jpg      
  inflating: img_testA/9658.jpg      
  inflating: img_testA/1204.jpg      
  inflating: img_testA/7675.jpg      
  inflating: img_testA/8546.jpg      
  inflating: img_testA/1210.jpg      
  inflating: img_testA/2719.jpg      
  inflating: img_testA/5076.jpg      
  inflating: img_testA/5710.jpg      
  inflating: img_testA/3361.jpg      
  inflating: img_testA/7107.jpg      
  inflating: img_testA/8234.jpg      
  inflating: img_testA/6219.jpg      
  inflating: img_testA/3349.jpg      
  inflating: img_testA/4426.jpg      
  inflating: img_testA/9102.jpg      
  inflating: img_testA/6557.jpg      
  inflating: img_testA/1238.jpg      
  inflating: img_testA/7649.jpg      
  inflating: img_testA/4340.jpg      
  inflating: img_testA/2731.jpg      
  inflating: img_testA/2725.jpg      
  inflating: img_testA/9670.jpg      
  inflating: img_testA/6543.jpg      
  inflating: img_testA/7885.jpg      
  inflating: img_testA/6225.jpg      
  inflating: img_testA/2043.jpg      
  inflating: img_testA/9499.jpg      
  inflating: img_testA/6972.jpg      
  inflating: img_testA/237.jpg       
  inflating: img_testA/4817.jpg      
  inflating: img_testA/6966.jpg      
  inflating: img_testA/6796.jpg      
  inflating: img_testA/7488.jpg      
  inflating: img_testA/2296.jpg      
  inflating: img_testA/1947.jpg      
  inflating: img_testA/4195.jpg      
  inflating: img_testA/8977.jpg      
  inflating: img_testA/8744.jpg      
  inflating: img_testA/7477.jpg      
  inflating: img_testA/1006.jpg      
  inflating: img_testA/6769.jpg      
  inflating: img_testA/5260.jpg      
  inflating: img_testA/4618.jpg      
  inflating: img_testA/1760.jpg      
  inflating: img_testA/1774.jpg      
  inflating: img_testA/8036.jpg      
  inflating: img_testA/3163.jpg      
  inflating: img_testA/5274.jpg      
  inflating: img_testA/3605.jpg      
  inflating: img_testA/8988.jpg      
  inflating: img_testA/2533.jpg      
  inflating: img_testA/4142.jpg      
  inflating: img_testA/9466.jpg      
  inflating: img_testA/9300.jpg      
  inflating: img_testA/4630.jpg      
  inflating: img_testA/7339.jpg      
  inflating: img_testA/6999.jpg      
  inflating: img_testA/9472.jpg      
  inflating: img_testA/2527.jpg      
  inflating: img_testA/4156.jpg      
  inflating: img_testA/8195.jpg      
  inflating: img_testA/4977.jpg      
  inflating: img_testA/2484.jpg      
  inflating: img_testA/3942.jpg      
  inflating: img_testA/4793.jpg      
  inflating: img_testA/6184.jpg      
  inflating: img_testA/6190.jpg      
  inflating: img_testA/5499.jpg      
  inflating: img_testA/8803.jpg      
  inflating: img_testA/1199.jpg      
  inflating: img_testA/1172.jpg      
  inflating: img_testA/8630.jpg      
  inflating: img_testA/3765.jpg      
  inflating: img_testA/5314.jpg      
  inflating: img_testA/5472.jpg      
  inflating: img_testA/3003.jpg      
  inflating: img_testA/1614.jpg      
  inflating: img_testA/7271.jpg      
  inflating: img_testA/1600.jpg      
  inflating: img_testA/394.jpg       
  inflating: img_testA/3771.jpg      
  inflating: img_testA/6609.jpg      
  inflating: img_testA/1166.jpg      
  inflating: img_testA/7517.jpg      
  inflating: img_testA/6621.jpg      
  inflating: img_testA/9512.jpg      
  inflating: img_testA/9274.jpg      
  inflating: img_testA/4988.jpg      
  inflating: img_testA/4750.jpg      
  inflating: img_testA/6153.jpg      
  inflating: img_testA/6635.jpg      
  inflating: img_testA/3995.jpg      
  inflating: img_testA/8397.jpg      
  inflating: img_testA/9089.jpg      
  inflating: img_testA/2862.jpg      
  inflating: img_testA/9923.jpg      
  inflating: img_testA/155.jpg       
  inflating: img_testA/7098.jpg      
  inflating: img_testA/4591.jpg      
  inflating: img_testA/5857.jpg      
  inflating: img_testA/3598.jpg      
  inflating: img_testA/7926.jpg      
  inflating: img_testA/83.jpg        
  inflating: img_testA/97.jpg        
  inflating: img_testA/4585.jpg      
  inflating: img_testA/169.jpg       
  inflating: img_testA/182.jpg       
  inflating: img_testA/7067.jpg      
  inflating: img_testA/3201.jpg      
  inflating: img_testA/2679.jpg      
  inflating: img_testA/1370.jpg      
  inflating: img_testA/8426.jpg      
  inflating: img_testA/9738.jpg      
  inflating: img_testA/5102.jpg      
  inflating: img_testA/3573.jpg      
  inflating: img_testA/5664.jpg      
  inflating: img_testA/196.jpg       
  inflating: img_testA/1402.jpg      
  inflating: img_testA/9076.jpg      
  inflating: img_testA/6345.jpg      
  inflating: img_testA/8368.jpg      
  inflating: img_testA/800.jpg       
  inflating: img_testA/9710.jpg      
  inflating: img_testA/40.jpg        
  inflating: img_testA/4234.jpg      
  inflating: img_testA/2651.jpg      
  inflating: img_testA/4220.jpg      
  inflating: img_testA/2889.jpg      
  inflating: img_testA/814.jpg       
  inflating: img_testA/7729.jpg      
  inflating: img_testA/54.jpg        
  inflating: img_testA/9062.jpg      
  inflating: img_testA/4546.jpg      
  inflating: img_testA/5658.jpg      
  inflating: img_testA/7077.jpg      
  inflating: img_testA/1406.jpg      
  inflating: img_testA/192.jpg       
  inflating: img_testA/3211.jpg      
  inflating: img_testA/3577.jpg      
  inflating: img_testA/4218.jpg      
  inflating: img_testA/5106.jpg      
  inflating: img_testA/2669.jpg      
  inflating: img_testA/1360.jpg      
  inflating: img_testA/8422.jpg      
  inflating: img_testA/9728.jpg      
  inflating: img_testA/8436.jpg      
  inflating: img_testA/7705.jpg      
  inflating: img_testA/5112.jpg      
  inflating: img_testA/5674.jpg      
  inflating: img_testA/3205.jpg      
  inflating: img_testA/8350.jpg      
  inflating: img_testA/1412.jpg      
  inflating: img_testA/2133.jpg      
  inflating: img_testA/9700.jpg      
  inflating: img_testA/810.jpg       
  inflating: img_testA/2899.jpg      
  inflating: img_testA/2641.jpg      
  inflating: img_testA/1348.jpg      
  inflating: img_testA/6427.jpg      
  inflating: img_testA/44.jpg        
  inflating: img_testA/7739.jpg      
  inflating: img_testA/6341.jpg      
  inflating: img_testA/2127.jpg      
  inflating: img_testA/4556.jpg      
  inflating: img_testA/9927.jpg      
  inflating: img_testA/2872.jpg      
  inflating: img_testA/2866.jpg      
  inflating: img_testA/9933.jpg      
  inflating: img_testA/8393.jpg      
  inflating: img_testA/7088.jpg      
  inflating: img_testA/5847.jpg      
  inflating: img_testA/7922.jpg      
  inflating: img_testA/87.jpg        
  inflating: img_testA/2682.jpg      
  inflating: img_testA/4595.jpg      
  inflating: img_testA/5853.jpg      
  inflating: img_testA/6382.jpg      
  inflating: img_testA/5304.jpg      
  inflating: img_testA/3775.jpg      
  inflating: img_testA/390.jpg       
  inflating: img_testA/8146.jpg      
  inflating: img_testA/1610.jpg      
  inflating: img_testA/3007.jpg      
  inflating: img_testA/5476.jpg      
  inflating: img_testA/3761.jpg      
  inflating: img_testA/6619.jpg      
  inflating: img_testA/2457.jpg      
  inflating: img_testA/5338.jpg      
  inflating: img_testA/3991.jpg      
  inflating: img_testA/4026.jpg      
  inflating: img_testA/1638.jpg      
  inflating: img_testA/7249.jpg      
  inflating: img_testA/4740.jpg      
  inflating: img_testA/4998.jpg      
  inflating: img_testA/2331.jpg      
  inflating: img_testA/4754.jpg      
  inflating: img_testA/2325.jpg      
  inflating: img_testA/9516.jpg      
  inflating: img_testA/2443.jpg      
  inflating: img_testA/4032.jpg      
  inflating: img_testA/4967.jpg      
  inflating: img_testA/8191.jpg      
  inflating: img_testA/8807.jpg      
  inflating: img_testA/3952.jpg      
  inflating: img_testA/1823.jpg      
  inflating: img_testA/1837.jpg      
  inflating: img_testA/6180.jpg      
  inflating: img_testA/5489.jpg      
  inflating: img_testA/1189.jpg      
  inflating: img_testA/1016.jpg      
  inflating: img_testA/7467.jpg      
  inflating: img_testA/3601.jpg      
  inflating: img_testA/2279.jpg      
  inflating: img_testA/5516.jpg      
  inflating: img_testA/1770.jpg      
  inflating: img_testA/8026.jpg      
  inflating: img_testA/9338.jpg      
  inflating: img_testA/5502.jpg      
  inflating: img_testA/3173.jpg      
  inflating: img_testA/8998.jpg      
  inflating: img_testA/596.jpg       
  inflating: img_testA/2523.jpg      
  inflating: img_testA/9476.jpg      
  inflating: img_testA/4634.jpg      
  inflating: img_testA/4620.jpg      
  inflating: img_testA/1758.jpg      
  inflating: img_testA/9462.jpg      
  inflating: img_testA/6751.jpg      
  inflating: img_testA/8797.jpg      
  inflating: img_testA/6962.jpg      
  inflating: img_testA/227.jpg       
  inflating: img_testA/4813.jpg      
  inflating: img_testA/4807.jpg      
  inflating: img_testA/233.jpg       
  inflating: img_testA/8783.jpg      
  inflating: img_testA/7498.jpg      
  inflating: img_testA/3826.jpg      
  inflating: img_testA/4191.jpg      
  inflating: img_testA/3198.jpg      
  inflating: img_testA/2286.jpg      
  inflating: img_testA/1943.jpg      
  inflating: img_testA/2292.jpg      
  inflating: img_testA/4185.jpg      
  inflating: img_testA/8967.jpg      
  inflating: img_testA/3365.jpg      
  inflating: img_testA/3403.jpg      
  inflating: img_testA/9890.jpg      
  inflating: img_testA/8556.jpg      
  inflating: img_testA/9648.jpg      
  inflating: img_testA/1214.jpg      
  inflating: img_testA/780.jpg       
  inflating: img_testA/2709.jpg      
  inflating: img_testA/5066.jpg      
  inflating: img_testA/8224.jpg      
  inflating: img_testA/7117.jpg      
  inflating: img_testA/3359.jpg      
  inflating: img_testA/5728.jpg      
  inflating: img_testA/2047.jpg      
  inflating: img_testA/7659.jpg      
  inflating: img_testA/6547.jpg      
  inflating: img_testA/9674.jpg      
  inflating: img_testA/7881.jpg      
  inflating: img_testA/4350.jpg      
  inflating: img_testA/2735.jpg      
  inflating: img_testA/4344.jpg      
  inflating: img_testA/970.jpg       
  inflating: img_testA/6553.jpg      
  inflating: img_testA/6235.jpg      
  inflating: img_testA/8218.jpg      
  inflating: img_testA/4422.jpg      
  inflating: img_testA/2053.jpg      
  inflating: img_testA/8595.jpg      
  inflating: img_testA/2912.jpg      
  inflating: img_testA/9847.jpg      
  inflating: img_testA/8581.jpg      
  inflating: img_testA/5933.jpg      
  inflating: img_testA/6584.jpg      
  inflating: img_testA/7842.jpg      
  inflating: img_testA/7856.jpg      
  inflating: img_testA/5927.jpg      
  inflating: img_testA/1599.jpg      
  inflating: img_testA/4386.jpg      
  inflating: img_testA/7857.jpg      
  inflating: img_testA/1598.jpg      
  inflating: img_testA/5926.jpg      
  inflating: img_testA/2091.jpg      
  inflating: img_testA/6585.jpg      
  inflating: img_testA/4392.jpg      
  inflating: img_testA/2913.jpg      
  inflating: img_testA/9852.jpg      
  inflating: img_testA/6552.jpg      
  inflating: img_testA/971.jpg       
  inflating: img_testA/4345.jpg      
  inflating: img_testA/2052.jpg      
  inflating: img_testA/4423.jpg      
  inflating: img_testA/8219.jpg      
  inflating: img_testA/9107.jpg      
  inflating: img_testA/6234.jpg      
  inflating: img_testA/9113.jpg      
  inflating: img_testA/6220.jpg      
  inflating: img_testA/4351.jpg      
  inflating: img_testA/9675.jpg      
  inflating: img_testA/6546.jpg      
  inflating: img_testA/4379.jpg      
  inflating: img_testA/2708.jpg      
  inflating: img_testA/795.jpg       
  inflating: img_testA/7670.jpg      
  inflating: img_testA/8543.jpg      
  inflating: img_testA/6208.jpg      
  inflating: img_testA/1567.jpg      
  inflating: img_testA/5701.jpg      
  inflating: img_testA/3370.jpg      
  inflating: img_testA/3364.jpg      
  inflating: img_testA/8231.jpg      
  inflating: img_testA/1573.jpg      
  inflating: img_testA/1215.jpg      
  inflating: img_testA/9649.jpg      
  inflating: img_testA/8557.jpg      
  inflating: img_testA/2293.jpg      
  inflating: img_testA/1942.jpg      
  inflating: img_testA/568.jpg       
  inflating: img_testA/8966.jpg      
  inflating: img_testA/4184.jpg      
  inflating: img_testA/4190.jpg      
  inflating: img_testA/3827.jpg      
  inflating: img_testA/6787.jpg      
  inflating: img_testA/7499.jpg      
  inflating: img_testA/1956.jpg      
  inflating: img_testA/3199.jpg      
  inflating: img_testA/232.jpg       
  inflating: img_testA/4806.jpg      
  inflating: img_testA/6977.jpg      
  inflating: img_testA/8782.jpg      
  inflating: img_testA/540.jpg       
  inflating: img_testA/226.jpg       
  inflating: img_testA/7328.jpg      
  inflating: img_testA/2250.jpg      
  inflating: img_testA/3628.jpg      
  inflating: img_testA/6988.jpg      
  inflating: img_testA/2244.jpg      
  inflating: img_testA/9311.jpg      
  inflating: img_testA/1995.jpg      
  inflating: img_testA/3172.jpg      
  inflating: img_testA/1765.jpg      
  inflating: img_testA/7314.jpg      
  inflating: img_testA/8741.jpg      
  inflating: img_testA/7472.jpg      
  inflating: img_testA/597.jpg       
  inflating: img_testA/1003.jpg      
  inflating: img_testA/5265.jpg      
  inflating: img_testA/3600.jpg      
  inflating: img_testA/8755.jpg      
  inflating: img_testA/7466.jpg      
  inflating: img_testA/583.jpg       
  inflating: img_testA/8033.jpg      
  inflating: img_testA/7300.jpg      
  inflating: img_testA/4609.jpg      
  inflating: img_testA/1188.jpg      
  inflating: img_testA/3953.jpg      
  inflating: img_testA/2495.jpg      
  inflating: img_testA/408.jpg       
  inflating: img_testA/1822.jpg      
  inflating: img_testA/4782.jpg      
  inflating: img_testA/8190.jpg      
  inflating: img_testA/4972.jpg      
  inflating: img_testA/6803.jpg      
  inflating: img_testA/420.jpg       
  inflating: img_testA/6817.jpg      
  inflating: img_testA/4966.jpg      
  inflating: img_testA/6142.jpg      
  inflating: img_testA/9271.jpg      
  inflating: img_testA/2324.jpg      
  inflating: img_testA/4033.jpg      
  inflating: img_testA/9517.jpg      
  inflating: img_testA/8609.jpg      
  inflating: img_testA/6630.jpg      
  inflating: img_testA/3748.jpg      
  inflating: img_testA/3990.jpg      
  inflating: img_testA/2456.jpg      
  inflating: img_testA/4999.jpg      
  inflating: img_testA/9265.jpg      
  inflating: img_testA/5477.jpg      
  inflating: img_testA/7260.jpg      
  inflating: img_testA/6618.jpg      
  inflating: img_testA/3012.jpg      
  inflating: img_testA/2683.jpg      
  inflating: img_testA/86.jpg        
  inflating: img_testA/7923.jpg      
  inflating: img_testA/5846.jpg      
  inflating: img_testA/6397.jpg      
  inflating: img_testA/7937.jpg      
  inflating: img_testA/622.jpg       
  inflating: img_testA/2867.jpg      
  inflating: img_testA/144.jpg       
  inflating: img_testA/805.jpg       
  inflating: img_testA/9715.jpg      
  inflating: img_testA/45.jpg        
  inflating: img_testA/1349.jpg      
  inflating: img_testA/4231.jpg      
  inflating: img_testA/5649.jpg      
  inflating: img_testA/9073.jpg      
  inflating: img_testA/6340.jpg      
  inflating: img_testA/8379.jpg      
  inflating: img_testA/3.jpg         
  inflating: __MACOSX/img_testA/._3.jpg  
  inflating: img_testA/2132.jpg      
  inflating: img_testA/4225.jpg      
  inflating: img_testA/9701.jpg      
  inflating: img_testA/51.jpg        
  inflating: img_testA/6432.jpg      
  inflating: img_testA/3562.jpg      
  inflating: img_testA/7704.jpg      
  inflating: img_testA/8437.jpg      
  inflating: img_testA/79.jpg        
  inflating: img_testA/9729.jpg      
  inflating: img_testA/1413.jpg      
  inflating: img_testA/187.jpg       
  inflating: img_testA/7062.jpg      
  inflating: img_testA/6368.jpg      
  inflating: img_testA/7710.jpg      
  inflating: img_testA/5107.jpg      
  inflating: img_testA/3576.jpg      
  inflating: img_testA/5677.jpg      
  inflating: img_testA/3206.jpg      
  inflating: img_testA/7060.jpg      
  inflating: img_testA/8353.jpg      
  inflating: img_testA/1411.jpg      
  inflating: img_testA/1377.jpg      
  inflating: img_testA/5111.jpg      
  inflating: img_testA/3574.jpg      
  inflating: img_testA/5105.jpg      
  inflating: img_testA/7712.jpg      
  inflating: img_testA/8347.jpg      
  inflating: img_testA/9071.jpg      
  inflating: img_testA/5893.jpg      
  inflating: img_testA/9717.jpg      
  inflating: img_testA/47.jpg        
  inflating: img_testA/6424.jpg      
  inflating: img_testA/807.jpg       
  inflating: img_testA/8409.jpg      
  inflating: img_testA/6430.jpg      
  inflating: img_testA/4227.jpg      
  inflating: img_testA/2656.jpg      
  inflating: img_testA/7048.jpg      
  inflating: img_testA/1439.jpg      
  inflating: img_testA/9930.jpg      
  inflating: img_testA/9924.jpg      
  inflating: img_testA/7909.jpg      
  inflating: img_testA/634.jpg       
  inflating: img_testA/2871.jpg      
  inflating: img_testA/152.jpg       
  inflating: img_testA/8384.jpg      
  inflating: img_testA/4596.jpg      
  inflating: img_testA/5850.jpg      
  inflating: img_testA/6381.jpg      
  inflating: img_testA/1388.jpg      
  inflating: img_testA/2859.jpg      
  inflating: img_testA/7935.jpg      
  inflating: img_testA/6395.jpg      
  inflating: img_testA/4582.jpg      
  inflating: img_testA/7504.jpg      
  inflating: img_testA/1175.jpg      
  inflating: img_testA/1613.jpg      
  inflating: img_testA/8151.jpg      
  inflating: img_testA/7262.jpg      
  inflating: img_testA/3004.jpg      
  inflating: img_testA/3010.jpg      
  inflating: img_testA/393.jpg       
  inflating: img_testA/1161.jpg      
  inflating: img_testA/2468.jpg      
  inflating: img_testA/3776.jpg      
  inflating: img_testA/9515.jpg      
  inflating: img_testA/2440.jpg      
  inflating: img_testA/2326.jpg      
  inflating: img_testA/9273.jpg      
  inflating: img_testA/6154.jpg      
  inflating: img_testA/9267.jpg      
  inflating: img_testA/8179.jpg      
  inflating: img_testA/2454.jpg      
  inflating: img_testA/6801.jpg      
  inflating: img_testA/3979.jpg      
  inflating: img_testA/1808.jpg      
  inflating: img_testA/344.jpg       
  inflating: img_testA/8186.jpg      
  inflating: img_testA/350.jpg       
  inflating: img_testA/8838.jpg      
  inflating: img_testA/436.jpg       
  inflating: img_testA/3945.jpg      
  inflating: img_testA/8810.jpg      
  inflating: img_testA/378.jpg       
  inflating: img_testA/6183.jpg      
  inflating: img_testA/4780.jpg      
  inflating: img_testA/1820.jpg      
  inflating: img_testA/6197.jpg      
  inflating: img_testA/3789.jpg      
  inflating: img_testA/3951.jpg      
  inflating: img_testA/5267.jpg      
  inflating: img_testA/8743.jpg      
  inflating: img_testA/5515.jpg      
  inflating: img_testA/581.jpg       
  inflating: img_testA/1015.jpg      
  inflating: img_testA/7464.jpg      
  inflating: img_testA/4145.jpg      
  inflating: img_testA/2534.jpg      
  inflating: img_testA/1983.jpg      
  inflating: img_testA/8019.jpg      
  inflating: img_testA/9307.jpg      
  inflating: img_testA/1997.jpg      
  inflating: img_testA/6020.jpg      
  inflating: img_testA/5529.jpg      
  inflating: img_testA/4637.jpg      
  inflating: img_testA/3158.jpg      
  inflating: img_testA/4151.jpg      
  inflating: img_testA/1029.jpg      
  inflating: img_testA/9475.jpg      
  inflating: img_testA/8780.jpg      
  inflating: img_testA/6975.jpg      
  inflating: img_testA/556.jpg       
  inflating: img_testA/230.jpg       
  inflating: img_testA/224.jpg       
  inflating: img_testA/1968.jpg      
  inflating: img_testA/542.jpg       
  inflating: img_testA/6949.jpg      
  inflating: img_testA/6791.jpg      
  inflating: img_testA/1798.jpg      
  inflating: img_testA/2291.jpg      
  inflating: img_testA/2285.jpg      
  inflating: img_testA/218.jpg       
  inflating: img_testA/1954.jpg      
  inflating: img_testA/8970.jpg      
  inflating: img_testA/3825.jpg      
  inflating: img_testA/1565.jpg      
  inflating: img_testA/9139.jpg      
  inflating: img_testA/9887.jpg      
  inflating: img_testA/797.jpg       
  inflating: img_testA/5071.jpg      
  inflating: img_testA/8555.jpg      
  inflating: img_testA/6578.jpg      
  inflating: img_testA/783.jpg       
  inflating: img_testA/1571.jpg      
  inflating: img_testA/3366.jpg      
  inflating: img_testA/4409.jpg      
  inflating: img_testA/5717.jpg      
  inflating: img_testA/9105.jpg      
  inflating: img_testA/6236.jpg      
  inflating: img_testA/7128.jpg      
  inflating: img_testA/4421.jpg      
  inflating: img_testA/2050.jpg      
  inflating: img_testA/5059.jpg      
  inflating: img_testA/3428.jpg      
  inflating: img_testA/973.jpg       
  inflating: img_testA/7896.jpg      
  inflating: img_testA/967.jpg       
  inflating: img_testA/8569.jpg      
  inflating: img_testA/9677.jpg      
  inflating: img_testA/7882.jpg      
  inflating: img_testA/2722.jpg      
  inflating: img_testA/4353.jpg      
  inflating: img_testA/9111.jpg      
  inflating: img_testA/6222.jpg      
  inflating: img_testA/5918.jpg      
  inflating: img_testA/7869.jpg      
  inflating: img_testA/754.jpg       
  inflating: img_testA/740.jpg       
  inflating: img_testA/9850.jpg      
  inflating: img_testA/8596.jpg      
  inflating: img_testA/998.jpg       
  inflating: img_testA/5924.jpg      
  inflating: img_testA/7855.jpg      
  inflating: img_testA/2939.jpg      
  inflating: img_testA/6587.jpg      
  inflating: img_testA/7841.jpg      
  inflating: img_testA/5930.jpg      
  inflating: img_testA/7698.jpg      
  inflating: img_testA/6586.jpg      
  inflating: img_testA/2938.jpg      
  inflating: img_testA/3398.jpg      
  inflating: img_testA/5925.jpg      
  inflating: img_testA/9879.jpg      
  inflating: img_testA/7854.jpg      
  inflating: img_testA/769.jpg       
  inflating: img_testA/6592.jpg      
  inflating: img_testA/8597.jpg      
  inflating: img_testA/999.jpg       
  inflating: img_testA/9851.jpg      
  inflating: img_testA/9689.jpg      
  inflating: img_testA/8583.jpg      
  inflating: img_testA/9845.jpg      
  inflating: img_testA/7868.jpg      
  inflating: img_testA/4352.jpg      
  inflating: img_testA/7883.jpg      
  inflating: img_testA/6545.jpg      
  inflating: img_testA/966.jpg       
  inflating: img_testA/8568.jpg      
  inflating: img_testA/6223.jpg      
  inflating: img_testA/2045.jpg      
  inflating: img_testA/4434.jpg      
  inflating: img_testA/2051.jpg      
  inflating: img_testA/4420.jpg      
  inflating: img_testA/6237.jpg      
  inflating: img_testA/9104.jpg      
  inflating: img_testA/1558.jpg      
  inflating: img_testA/4346.jpg      
  inflating: img_testA/1216.jpg      
  inflating: img_testA/6579.jpg      
  inflating: img_testA/8554.jpg      
  inflating: img_testA/7667.jpg      
  inflating: img_testA/9892.jpg      
  inflating: img_testA/5070.jpg      
  inflating: img_testA/5716.jpg      
  inflating: img_testA/9138.jpg      
  inflating: img_testA/3415.jpg      
  inflating: img_testA/5064.jpg      
  inflating: img_testA/1202.jpg      
  inflating: img_testA/796.jpg       
  inflating: img_testA/8540.jpg      
  inflating: img_testA/1955.jpg      
  inflating: img_testA/219.jpg       
  inflating: img_testA/4193.jpg      
  inflating: img_testA/6784.jpg      
  inflating: img_testA/6790.jpg      
  inflating: img_testA/6948.jpg      
  inflating: img_testA/4187.jpg      
  inflating: img_testA/5299.jpg      
  inflating: img_testA/3830.jpg      
  inflating: img_testA/2290.jpg      
  inflating: img_testA/4811.jpg      
  inflating: img_testA/1969.jpg      
  inflating: img_testA/6960.jpg      
  inflating: img_testA/3818.jpg      
  inflating: img_testA/557.jpg       
  inflating: img_testA/8781.jpg      
  inflating: img_testA/231.jpg       
  inflating: img_testA/4805.jpg      
  inflating: img_testA/3159.jpg      
  inflating: img_testA/2247.jpg      
  inflating: img_testA/6021.jpg      
  inflating: img_testA/1996.jpg      
  inflating: img_testA/6035.jpg      
  inflating: img_testA/2253.jpg      
  inflating: img_testA/1772.jpg      
  inflating: img_testA/7303.jpg      
  inflating: img_testA/3165.jpg      
  inflating: img_testA/5272.jpg      
  inflating: img_testA/9448.jpg      
  inflating: img_testA/1000.jpg      
  inflating: img_testA/5266.jpg      
  inflating: img_testA/4178.jpg      
  inflating: img_testA/3617.jpg      
  inflating: img_testA/3171.jpg      
  inflating: img_testA/6009.jpg      
  inflating: img_testA/7317.jpg      
  inflating: img_testA/6196.jpg      
  inflating: img_testA/1821.jpg      
  inflating: img_testA/4781.jpg      
  inflating: img_testA/4959.jpg      
  inflating: img_testA/3950.jpg      
  inflating: img_testA/8805.jpg      
  inflating: img_testA/3944.jpg      
  inflating: img_testA/2482.jpg      
  inflating: img_testA/4795.jpg      
  inflating: img_testA/1835.jpg      
  inflating: img_testA/4965.jpg      
  inflating: img_testA/351.jpg       
  inflating: img_testA/3978.jpg      
  inflating: img_testA/6800.jpg      
  inflating: img_testA/423.jpg       
  inflating: img_testA/345.jpg       
  inflating: img_testA/4971.jpg      
  inflating: img_testA/2333.jpg      
  inflating: img_testA/4742.jpg      
  inflating: img_testA/8178.jpg      
  inflating: img_testA/9266.jpg      
  inflating: img_testA/9500.jpg      
  inflating: img_testA/3987.jpg      
  inflating: img_testA/1148.jpg      
  inflating: img_testA/9514.jpg      
  inflating: img_testA/9272.jpg      
  inflating: img_testA/6141.jpg      
  inflating: img_testA/5448.jpg      
  inflating: img_testA/4756.jpg      
  inflating: img_testA/8144.jpg      
  inflating: img_testA/6169.jpg      
  inflating: img_testA/1606.jpg      
  inflating: img_testA/3777.jpg      
  inflating: img_testA/1160.jpg      
  inflating: img_testA/8622.jpg      
  inflating: img_testA/9528.jpg      
  inflating: img_testA/8636.jpg      
  inflating: img_testA/3763.jpg      
  inflating: img_testA/1612.jpg      
  inflating: img_testA/91.jpg        
  inflating: img_testA/4583.jpg      
  inflating: img_testA/6394.jpg      
  inflating: img_testA/6380.jpg      
  inflating: img_testA/5689.jpg      
  inflating: img_testA/5851.jpg      
  inflating: img_testA/2858.jpg      
  inflating: img_testA/2680.jpg      
  inflating: img_testA/85.jpg        
  inflating: img_testA/2870.jpg      
  inflating: img_testA/635.jpg       
  inflating: img_testA/8385.jpg      
  inflating: img_testA/153.jpg       
  inflating: img_testA/5879.jpg      
  inflating: img_testA/8391.jpg      
  inflating: img_testA/621.jpg       
  inflating: img_testA/2864.jpg      
  inflating: img_testA/4226.jpg      
  inflating: img_testA/3549.jpg      
  inflating: img_testA/812.jpg       
  inflating: img_testA/52.jpg        
  inflating: img_testA/9064.jpg      
  inflating: img_testA/7049.jpg      
  inflating: img_testA/5886.jpg      
  inflating: img_testA/4540.jpg      
  inflating: img_testA/4554.jpg      
  inflating: img_testA/2125.jpg      
  inflating: img_testA/806.jpg       
  inflating: img_testA/8408.jpg      
  inflating: img_testA/46.jpg        
  inflating: img_testA/9716.jpg      
  inflating: img_testA/2643.jpg      
  inflating: img_testA/8420.jpg      
  inflating: img_testA/1362.jpg      
  inflating: img_testA/5104.jpg      
  inflating: img_testA/190.jpg       
  inflating: img_testA/1404.jpg      
  inflating: img_testA/184.jpg       
  inflating: img_testA/8352.jpg      
  inflating: img_testA/3207.jpg      
  inflating: img_testA/5676.jpg      
  inflating: img_testA/2119.jpg      
  inflating: img_testA/5110.jpg      
  inflating: img_testA/3561.jpg      
  inflating: img_testA/8434.jpg      
  inflating: img_testA/7707.jpg      
  inflating: img_testA/1376.jpg
登录后复制
       

为了监控训练,便于保存最佳模型,我们对数据进行随机划分训练集和验证集,由于数据量较多,验证集图片数量划分少一些,按照0.97:0.03划分

In [ ]
import matplotlib.pyplot as pltimport numpy as npimport osimport cv2import randomimport shutilfrom shutil import copyfile
data_path = "img_train//"path_origin = "img_train"label_path = "lab_train"files = list(filter(lambda x: x.endswith('.jpg'), os.listdir(path_origin)))
random.shuffle(files)
rate = int(len(files) * 0.97)
train_txt = open('train.txt','w')
val_txt = open('val.txt','w')for i,f in enumerate(files):
    image_path = os.path.join(path_origin, f)
    label_name = f.split('.')[0]+".png"
    target_image = os.path.join(path_origin, f)
    target_label = os.path.join(label_path, label_name)    if i < rate: 
        train_txt.write(target_image + ' ' + target_label+ '\n')    else:
        val_txt.write(target_image + ' ' + target_label+ '\n')
登录后复制
   
In [ ]
files = list(filter(lambda x: x.endswith('.jpg'), os.listdir("img_testA/")))print(len(files))
登录后复制
       
4608
登录后复制
       

所选用的模型

在本次任务中我们选择语义分割SOTA网络结构OCRNet OCRNet着眼于语义分割中的上下文聚合问题提出了一种简单而有效的方法,即对象上下文表示,通过利用相应对象类的表示来表征像素。首先,在地面真值分割的监督下学习目标区域。其次,通过聚集对象区域中像素的表示来计算对象区域的表示。最后,计算每个像素和每个目标区域之间的关系,并用对象上下文表示来增强每个像素的表示,这是所有对象区域表示的加权聚合。网络结构如下图:

飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网        

backbone我们选择HRNet

飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网        

模型训练与调优

使用PaddleSeg中ocrnet进行训练,具体训练配置可见configs/ocrnet/remote.yml

batch_size: 64

iters: 1600000

train_dataset:

type: Dataset

飞桨PaddlePaddle
飞桨PaddlePaddle

飞桨PaddlePaddle开发者社区与布道,与社区共同进步

飞桨PaddlePaddle12
查看详情 飞桨PaddlePaddle

dataset_root: /home/aistudio

train_path: /home/aistudio/train.txt

num_classes: 4

transforms:

- type: ResizeStepScaling
  min_scale_factor: 0.5
  max_scale_factor: 2.0
  scale_step_size: 0.25
- type: RandomPaddingCrop
  crop_size: [256, 256]
- type: RandomHorizontalFlip
- type: Normalize
登录后复制
       

mode: train

val_dataset:

type: Dataset

dataset_root: /home/aistudio

val_path: /home/aistudio/val.txt

num_classes: 4

transforms:

- type: Normalize
登录后复制
       

mode: val

model:

type: OCRNet

backbone:

type: HRNet_W18

pretrained: https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz
登录后复制
       

backbone_indices: [0]

optimizer:

type: sgd

learning_rate:

value: 0.01

decay:

type: poly

power: 0.9
登录后复制
       

loss:

types:

- type: MixedLoss

  losses:
  
    - type: CrossEntropyLoss
    
    - type: LovaszHingeLoss
    
  coef: [1, 0.01]
  
- type: MixedLoss

  losses:
  
    - type: CrossEntropyLoss
    
    - type: LovaszHingeLoss
    
  coef: [1, 0.01]
登录后复制
       

coef: [1, 0.4]

损失函数使用 CrossEntropyLoss和LovaszHingeLoss的联合损失,改善像素类别不平衡的问题

In [ ]
cd PaddleSeg/
登录后复制
       
/home/aistudio/PaddleSeg
登录后复制
       

训练

In [ ]
!python train.py --config configs/ocrnet/remote.yml --do_eval #--resume_model output/best_model
登录后复制
   

进行测试

由于在paddleseg没有找到保存图片的选项,所以对源代码添加了图片保存,具体如下130-133行,添加文件 PaddleSeg/paddleseg/core/predict.py

飞桨常规赛:遥感影像地块分割 - 4月第8名方案 - php中文网        

In [7]
!python predict.py --config configs/ocrnet/remote.yml --model_path output/best_model/model.pdparams --image_path ../img_testA/
登录后复制
       
/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/layers/utils.py:26: DeprecationWarning: `np.int` is a deprecated alias for the builtin `int`. To silence this warning, use `int` by itself. Doing this will not modify any behavior and is safe. When replacing `np.int`, you may wish to use e.g. `np.int64` or `np.int32` to specify the precision. If you wish to review your current use, check the release note link for additional information.
Deprecated in NumPy 1.20; for more details and guidance: https://numpy.org/devdocs/release/1.20.0-notes.html#deprecations
  def convert_to_list(value, n, name, dtype=np.int):
2021-05-09 08:37:17 [INFO]
---------------Config Information---------------
batch_size: 64
iters: 1600000
learning_rate:
  decay:
    power: 0.9
    type: poly
  value: 0.01
loss:
  coef:
  - 1
  - 0.4
  types:
  - coef:
    - 1
    - 0.01
    losses:
    - type: CrossEntropyLoss
    - type: LovaszHingeLoss
    type: MixedLoss
  - coef:
    - 1
    - 0.01
    losses:
    - type: CrossEntropyLoss
    - type: LovaszHingeLoss
    type: MixedLoss
model:
  backbone:
    pretrained: https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz
    type: HRNet_W18
  backbone_indices:
  - 0
  type: OCRNet
optimizer:
  type: sgd
train_dataset:
  dataset_root: /home/aistudio
  mode: train
  num_classes: 4
  train_path: /home/aistudio/train.txt
  transforms:
  - max_scale_factor: 2.0
    min_scale_factor: 0.5
    scale_step_size: 0.25
    type: ResizeStepScaling
  - crop_size:
    - 256
    - 256
    type: RandomPaddingCrop
  - type: RandomHorizontalFlip
  - type: Normalize
  type: Dataset
val_dataset:
  dataset_root: /home/aistudio
  mode: val
  num_classes: 4
  transforms:
  - type: Normalize
  type: Dataset
  val_path: /home/aistudio/val.txt
------------------------------------------------
W0509 08:37:18.013245  3052 device_context.cc:362] Please NOTE: device: 0, GPU Compute Capability: 7.0, Driver API Version: 10.1, Runtime API Version: 10.1
W0509 08:37:18.013303  3052 device_context.cc:372] device: 0, cuDNN Version: 7.6.
2021-05-09 08:37:23 [INFO]	Loading pretrained model from https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz
2021-05-09 08:37:23,275 - INFO - Lock 139755317736016 acquired on /home/aistudio/.paddleseg/tmp/hrnet_w18_ssld
2021-05-09 08:37:23,275 - INFO - Lock 139755317736016 released on /home/aistudio/.paddleseg/tmp/hrnet_w18_ssld
2021-05-09 08:37:24 [INFO]	There are 1525/1525 variables loaded into HRNet.
2021-05-09 08:37:24 [INFO]	Number of predict images = 4608
2021-05-09 08:37:24 [INFO]	Loading pretrained model from output/best_model/model.pdparams
2021-05-09 08:37:25 [INFO]	There are 1583/1583 variables loaded into OCRNet.
2021-05-09 08:37:25 [INFO]	Start to predict...
 227/4608 [>.............................] - ETA: 8:3^C
Traceback (most recent call last):
  File "predict.py", line 177, in <module>
    main(args)
  File "predict.py", line 171, in main
    stride=args.stride,
  File "/home/aistudio/PaddleSeg/paddleseg/core/predict.py", line 115, in predict
    crop_size=crop_size)
  File "/home/aistudio/PaddleSeg/paddleseg/core/infer.py", line 186, in inference
    logits = model(im)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 902, in __call__
    outputs = self.forward(*inputs, **kwargs)
  File "/home/aistudio/PaddleSeg/paddleseg/models/ocrnet.py", line 71, in forward
    feats = self.backbone(x)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 902, in __call__
    outputs = self.forward(*inputs, **kwargs)
  File "/home/aistudio/PaddleSeg/paddleseg/models/backbones/hrnet.py", line 169, in forward
    st4 = self.st4(tr3)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 902, in __call__
    outputs = self.forward(*inputs, **kwargs)
  File "/home/aistudio/PaddleSeg/paddleseg/models/backbones/hrnet.py", line 506, in forward
    out = self.stage_func_list[idx](out)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 902, in __call__
    outputs = self.forward(*inputs, **kwargs)
  File "/home/aistudio/PaddleSeg/paddleseg/models/backbones/hrnet.py", line 536, in forward
    out = self.branches_func(x)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 902, in __call__
    outputs = self.forward(*inputs, **kwargs)
  File "/home/aistudio/PaddleSeg/paddleseg/models/backbones/hrnet.py", line 297, in forward
    conv = basic_block_func(conv)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 902, in __call__
    outputs = self.forward(*inputs, **kwargs)
  File "/home/aistudio/PaddleSeg/paddleseg/models/backbones/hrnet.py", line 413, in forward
    conv1 = self.conv1(x)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/dygraph/layers.py", line 886, in __call__
    with param_guard(self._parameters), param_guard(self._buffers):
  File "</opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/decorator.py:decorator-gen-40>", line 2, in param_guard
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/site-packages/paddle/fluid/wrapped_decorator.py", line 24, in __impl__
    wrapped_func = decorator_func(func)
  File "/opt/conda/envs/python35-paddle120-env/lib/python3.7/contextlib.py", line 237, in contextmanager
    @wraps(func)
KeyboardInterrupt
登录后复制
       

压缩结果

In [10]
!zip result.zip -r output/result/result/
登录后复制
   

以上就是飞桨常规赛:遥感影像地块分割 - 4月第8名方案的详细内容,更多请关注php中文网其它相关文章!

相关标签:
最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习
PHP中文网抖音号
发现有趣的

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号