基于Pytorch的UNet语义分割模型与代码

上传者: 19865329 | 上传时间: 2023-03-10 17:36:33 | 文件大小: 608.14MB | 文件类型: 7Z
模型在FloodNet数据集上进行了训练,mIOU在0.83左右,可训练自己的数据集 建议在训练网络的时候将输入的训练集其切分为384x384的小图片后,再来进行训练 模型采用标准的UNet,可以采用如下方式训练你自己的模型 数据集地址可以在train.py中修改为你自己的文件夹 python train.py -- --epochs 20 --batch-size 16 --learning-rate 2e-4 --scale 1 --validation 0.1 --classes 10 --amp 其中--amp为半精度训练 --scale是训练的时候对图片进行缩放,已经裁剪为384x384后就不需要再裁剪了

文件下载

资源详情

[{"title":"( 632 个子文件 608.14MB ) 基于Pytorch的UNet语义分割模型与代码","children":[{"title":"evaluate.py <span style='color:#111;'> 1.80KB </span>","children":null,"spread":false},{"title":"main.yml <span style='color:#111;'> 1.13KB </span>","children":null,"spread":false},{"title":".keep <span style='color:#111;'> 0B </span>","children":null,"spread":false},{"title":".keep <span style='color:#111;'> 0B </span>","children":null,"spread":false},{"title":"LICENSE <span style='color:#111;'> 34.32KB </span>","children":null,"spread":false},{"title":"hubconf.py <span style='color:#111;'> 911B </span>","children":null,"spread":false},{"title":"predict.py <span style='color:#111;'> 4.36KB </span>","children":null,"spread":false},{"title":"__init__.py <span style='color:#111;'> 29B </span>","children":null,"spread":false},{"title":"unet_parts.py <span style='color:#111;'> 2.54KB </span>","children":null,"spread":false},{"title":"unet_model.py <span style='color:#111;'> 1.79KB </span>","children":null,"spread":false},{"title":"unet_model.cpython-38.pyc <span style='color:#111;'> 1.71KB </span>","children":null,"spread":false},{"title":"unet_parts.cpython-38.pyc <span style='color:#111;'> 2.81KB </span>","children":null,"spread":false},{"title":"__init__.cpython-38.pyc <span style='color:#111;'> 210B </span>","children":null,"spread":false},{"title":"unet_parts-checkpoint.py <span style='color:#111;'> 2.54KB </span>","children":null,"spread":false},{"title":"unet_model-checkpoint.py <span style='color:#111;'> 1.79KB </span>","children":null,"spread":false},{"title":"utils.py <span style='color:#111;'> 531B </span>","children":null,"spread":false},{"title":"__init__.py <span style='color:#111;'> 0B </span>","children":null,"spread":false},{"title":"data_loading.py <span style='color:#111;'> 4.61KB </span>","children":null,"spread":false},{"title":"dice_score.py <span style='color:#111;'> 1.13KB </span>","children":null,"spread":false},{"title":"utils.cpython-38.pyc <span style='color:#111;'> 737B </span>","children":null,"spread":false},{"title":"dice_score.cpython-38.pyc <span style='color:#111;'> 1.09KB </span>","children":null,"spread":false},{"title":"data_loading.cpython-38.pyc <span style='color:#111;'> 4.83KB </span>","children":null,"spread":false},{"title":"__init__.cpython-38.pyc <span style='color:#111;'> 172B </span>","children":null,"spread":false},{"title":"data_loading-checkpoint.py <span style='color:#111;'> 4.61KB </span>","children":null,"spread":false},{"title":"dice_score-checkpoint.py <span style='color:#111;'> 1.13KB </span>","children":null,"spread":false},{"title":"utils-checkpoint.py <span style='color:#111;'> 531B </span>","children":null,"spread":false},{"title":"wandb-summary.json <span style='color:#111;'> 224.57KB </span>","children":null,"spread":false},{"title":"true_20873_61faeb8b52a8d76b0b1f.png <span style='color:#111;'> 1.26KB </span>","children":null,"spread":false},{"title":"pred_48208_9cc9b4b046ec7d075b4b.png <span style='color:#111;'> 1.99KB </span>","children":null,"spread":false},{"title":"true_27334_cf1814bcb8d13495b710.png <span style='color:#111;'> 509B </span>","children":null,"spread":false},{"title":"images_28825_e111b5faf2985eb5acfe.png <span style='color:#111;'> 211.60KB </span>","children":null,"spread":false},{"title":"images_22364_3eb3867da8dc0f32ac0f.png <span style='color:#111;'> 199.43KB </span>","children":null,"spread":false},{"title":"pred_43238_6e5743e1e31401c82a29.png <span style='color:#111;'> 5.01KB </span>","children":null,"spread":false},{"title":"images_12921_637b77eab183f9657fd7.png <span style='color:#111;'> 183.36KB </span>","children":null,"spread":false},{"title":"images_24849_b736620bd1002ada204d.png <span style='color:#111;'> 271.88KB </span>","children":null,"spread":false},{"title":"true_12921_8a170680264aabb17562.png <span style='color:#111;'> 1.88KB </span>","children":null,"spread":false},{"title":"true_43735_3c3336bb9e4f56d9f919.png <span style='color:#111;'> 1.62KB </span>","children":null,"spread":false},{"title":"true_29819_c4793ec3363d408e20d9.png <span style='color:#111;'> 2.00KB </span>","children":null,"spread":false},{"title":"pred_39262_ebe30144dd9bb51c7b67.png <span style='color:#111;'> 3.63KB </span>","children":null,"spread":false},{"title":"pred_7951_d41bd65bd2da22ae9927.png <span style='color:#111;'> 1.98KB </span>","children":null,"spread":false},{"title":"images_2981_eaa0608142c44aef7d22.png <span style='color:#111;'> 129.06KB </span>","children":null,"spread":false},{"title":"true_25346_743b439e73990f675855.png <span style='color:#111;'> 4.36KB </span>","children":null,"spread":false},{"title":"images_36280_cd6cd648762ff1d1f556.png <span style='color:#111;'> 4.02KB </span>","children":null,"spread":false},{"title":"images_6460_83f3092de2e2bdb90bb6.png <span style='color:#111;'> 141.80KB </span>","children":null,"spread":false},{"title":"pred_49202_bdff5d38de6774745ae9.png <span style='color:#111;'> 5.42KB </span>","children":null,"spread":false},{"title":"pred_3975_4959fe917e687f5ed65f.png <span style='color:#111;'> 5.49KB </span>","children":null,"spread":false},{"title":"images_15406_b620f4ef5b3f1078b5aa.png <span style='color:#111;'> 225.99KB </span>","children":null,"spread":false},{"title":"images_6957_02dcac1ccdd5b973da91.png <span style='color:#111;'> 231.07KB </span>","children":null,"spread":false},{"title":"pred_46717_98c7d67339a71f24a34d.png <span style='color:#111;'> 5.72KB </span>","children":null,"spread":false},{"title":"true_15406_bab28d2ae1740c87e761.png <span style='color:#111;'> 1.43KB </span>","children":null,"spread":false},{"title":"true_36777_29a53a093dfb87899435.png <span style='color:#111;'> 4.46KB </span>","children":null,"spread":false},{"title":"images_49699_4b7be9a2f5879eb75377.png <span style='color:#111;'> 269.79KB </span>","children":null,"spread":false},{"title":"pred_40753_7c4264b2fd3d190c07f6.png <span style='color:#111;'> 4.97KB </span>","children":null,"spread":false},{"title":"true_11927_2ff926fa867fa147093d.png <span style='color:#111;'> 1.56KB </span>","children":null,"spread":false},{"title":"true_9939_61faeb8b52a8d76b0b1f.png <span style='color:#111;'> 1.26KB </span>","children":null,"spread":false},{"title":"pred_40256_d5178edacbaff0a2bd65.png <span style='color:#111;'> 1.51KB </span>","children":null,"spread":false},{"title":"pred_29322_41265a1500ab99fde83e.png <span style='color:#111;'> 2.25KB </span>","children":null,"spread":false},{"title":"pred_28825_9fa19c9b9b3c6cbb502d.png <span style='color:#111;'> 2.64KB </span>","children":null,"spread":false},{"title":"images_18388_8cceead7c6e2306a5d54.png <span style='color:#111;'> 317.28KB </span>","children":null,"spread":false},{"title":"pred_32801_f149eb95530632baaa07.png <span style='color:#111;'> 2.17KB </span>","children":null,"spread":false},{"title":"pred_25346_dccbb38f7293980beb98.png <span style='color:#111;'> 4.53KB </span>","children":null,"spread":false},{"title":"pred_23855_9f4d23f714784f3a1a65.png <span style='color:#111;'> 5.24KB </span>","children":null,"spread":false},{"title":"pred_19382_5523eec50e39bad9041d.png <span style='color:#111;'> 1.70KB </span>","children":null,"spread":false},{"title":"pred_41250_e181ec57e1b581061995.png <span style='color:#111;'> 5.40KB </span>","children":null,"spread":false},{"title":"pred_5963_9e038f8b947319746b15.png <span style='color:#111;'> 1.75KB </span>","children":null,"spread":false},{"title":"true_16400_c0330f1ced841c175a12.png <span style='color:#111;'> 2.83KB </span>","children":null,"spread":false},{"title":"true_34292_4474351d9968348dff0b.png <span style='color:#111;'> 2.42KB </span>","children":null,"spread":false},{"title":"pred_17394_7c030423031e2a26359a.png <span style='color:#111;'> 3.68KB </span>","children":null,"spread":false},{"title":"pred_36280_da60ea56c1c21ed31a4c.png <span style='color:#111;'> 1.28KB </span>","children":null,"spread":false},{"title":"images_19382_d5ddcc53fce98c0794bf.png <span style='color:#111;'> 276.56KB </span>","children":null,"spread":false},{"title":"true_49699_c850b16a804de94a818a.png <span style='color:#111;'> 3.65KB </span>","children":null,"spread":false},{"title":"pred_44729_0b53c4b1dd09ee063cd0.png <span style='color:#111;'> 5.04KB </span>","children":null,"spread":false},{"title":"true_10933_cf1814bcb8d13495b710.png <span style='color:#111;'> 509B </span>","children":null,"spread":false},{"title":"true_41250_380dada958f943356d8d.png <span style='color:#111;'> 1.27KB </span>","children":null,"spread":false},{"title":"true_496_df37f7c63042c3d25373.png <span style='color:#111;'> 1.79KB </span>","children":null,"spread":false},{"title":"images_35286_94e780a3619eac177bef.png <span style='color:#111;'> 295.94KB </span>","children":null,"spread":false},{"title":"images_43735_f9bfb6e0a8c7e0cc7e93.png <span style='color:#111;'> 145.31KB </span>","children":null,"spread":false},{"title":"true_993_c3f4bfff6384d65249c5.png <span style='color:#111;'> 2.73KB </span>","children":null,"spread":false},{"title":"pred_14909_4c954502126940207280.png <span style='color:#111;'> 6.44KB </span>","children":null,"spread":false},{"title":"pred_4472_eec3362158d687cad122.png <span style='color:#111;'> 1.97KB </span>","children":null,"spread":false},{"title":"images_11430_9da23a82f64958357314.png <span style='color:#111;'> 302.26KB </span>","children":null,"spread":false},{"title":"pred_39759_f193f7e09a513dc78cea.png <span style='color:#111;'> 1.30KB </span>","children":null,"spread":false},{"title":"true_22861_644df6ae8c183cc9109e.png <span style='color:#111;'> 3.08KB </span>","children":null,"spread":false},{"title":"pred_8945_9a61de51919a0f416445.png <span style='color:#111;'> 1.94KB </span>","children":null,"spread":false},{"title":"images_34789_c1bad0f6b0a81e81cf78.png <span style='color:#111;'> 161.79KB </span>","children":null,"spread":false},{"title":"true_32801_ad07fe16c3e0cc68cd26.png <span style='color:#111;'> 1.64KB </span>","children":null,"spread":false},{"title":"pred_27831_df29328938244df1802e.png <span style='color:#111;'> 7.21KB </span>","children":null,"spread":false},{"title":"true_22364_fa850f623fad3345271f.png <span style='color:#111;'> 981B </span>","children":null,"spread":false},{"title":"pred_32304_69ae7c2ce03e82bb4e46.png <span style='color:#111;'> 1.27KB </span>","children":null,"spread":false},{"title":"true_30316_3cb560510c1a2036b1f8.png <span style='color:#111;'> 2.34KB </span>","children":null,"spread":false},{"title":"pred_34789_3d608a04e845d25b6130.png <span style='color:#111;'> 1.50KB </span>","children":null,"spread":false},{"title":"true_34789_972c3aa0452765b7237a.png <span style='color:#111;'> 1.29KB </span>","children":null,"spread":false},{"title":"images_20376_615683393f1af2ea3d84.png <span style='color:#111;'> 138.80KB </span>","children":null,"spread":false},{"title":"images_37274_c9627e73c2f8f96f9246.png <span style='color:#111;'> 198.20KB </span>","children":null,"spread":false},{"title":"pred_21867_8b510b1b16bf1b1c912e.png <span style='color:#111;'> 1.37KB </span>","children":null,"spread":false},{"title":"true_25843_cf1814bcb8d13495b710.png <span style='color:#111;'> 509B </span>","children":null,"spread":false},{"title":"images_45226_8498f87d5dafbfdd3fd0.png <span style='color:#111;'> 340.48KB </span>","children":null,"spread":false},{"title":"true_6460_84752424d0a04c6e907c.png <span style='color:#111;'> 3.13KB </span>","children":null,"spread":false},{"title":"pred_28328_ee0649c7ad18e6a96e8e.png <span style='color:#111;'> 1.37KB </span>","children":null,"spread":false},{"title":"pred_34292_88297d25e705c245b86f.png <span style='color:#111;'> 2.17KB </span>","children":null,"spread":false},{"title":"......","children":null,"spread":false},{"title":"<span style='color:steelblue;'>文件过多,未全部展示</span>","children":null,"spread":false}],"spread":true}]

评论信息

免责申明

【只为小站】的资源来自网友分享,仅供学习研究,请务必在下载后24小时内给予删除,不得用于其他任何用途,否则后果自负。基于互联网的特殊性,【只为小站】 无法对用户传输的作品、信息、内容的权属或合法性、合规性、真实性、科学性、完整权、有效性等进行实质审查;无论 【只为小站】 经营者是否已进行审查,用户均应自行承担因其传输的作品、信息、内容而可能或已经产生的侵权或权属纠纷等法律责任。
本站所有资源不代表本站的观点或立场,基于网友分享,根据中国法律《信息网络传播权保护条例》第二十二条之规定,若资源存在侵权或相关问题请联系本站客服人员,zhiweidada#qq.com,请把#换成@,本站将给予最大的支持与配合,做到及时反馈和处理。关于更多版权及免责申明参见 版权及免责申明