Github复现-测试基于transformer的变化检测模型BIT_CD

这篇具有很好参考价值的文章主要介绍了Github复现-测试基于transformer的变化检测模型BIT_CD。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。

源码链接: 

GitHub - justchenhao/BIT_CD: Official Pytorch Implementation of "Remote Sensing Image Change Detection with Transformers"Official Pytorch Implementation of "Remote Sensing Image Change Detection with Transformers" - GitHub - justchenhao/BIT_CD: Official Pytorch Implementation of "Remote Sensing Image Change Detection with Transformers"https://github.com/justchenhao/BIT_CD这个复现很简单,模型效果在公共数据集还不错,下面简单说下我测试的过程

1.数据说明

数据下载链接:

LEVIR-CD | LEVIR-CD is a new large-scale remote sensing binary change detection dataset, which would help develop novel deep learning-based algorithms for remote sensing image change detection.LEVIR-CD is a new large-scale remote sensing binary change detection dataset, which would help develop novel deep learning-based algorithms for remote sensing image change detection.https://justchenhao.github.io/LEVIR/里面有百度网盘的链接,下载出问题可以去我上一篇变化检测的复现,里面有下载的较详细说明。

2.数据结构

bit_cd,transformer,Transformer,深度学习

A里面放了第一期影像的训练、验证、测试数据,B里面放了 第二期影像的训练、验证、测试数据,label里面是标签,list 里面放了train.txt、val.txt、test.txt

bit_cd,transformer,Transformer,深度学习bit_cd,transformer,Transformer,深度学习

上面的结构意思就是所有数据放一起,通过写txt来控制训练、验证、预测图像 

 3.训练

bit_cd,transformer,Transformer,深度学习

 说明:

1.gpu_ids GPU id ,一个显卡就默认是0

2.project_name  项目名称,每次跑整个不一样的名字,整个名字后面预测的时候要对应起来

3.checkpoint_root  路径,放置模型,训练过程记录,最终的eval结构

bit_cd,transformer,Transformer,深度学习

4.num_workers , 加载数据的线程可以在utils.py中加载数据的地方看到

5.dataset,  数据集名字,这个目前只有一个,可以自己加新的

6.data_name 这个参数在项目里有点绕,这个默认得不要改就行,先看utils.py就发现,这个参数关系到训练数据的路径,接着看data_config.py文件发现,训练集的根目录要自己给出

bit_cd,transformer,Transformer,深度学习 bit_cd,transformer,Transformer,深度学习

 7.batch_size 默认

8.split, split_val 这两个默认就行

9.img_size  图像会被resize

10.类别默认

后面的参数就不说了,应该都认识

改完上面的参数就可以直接训练了

4.预测

预测用demo.py这个文件,文件的逻辑和训练的逻辑一致,参数基本都要和训练保持一致,那个output_folder可以换成存放结果的地方

bit_cd,transformer,Transformer,深度学习

 用来验证效果的测试数据放在了quik_start控制的路径下,数据结构和训练数据结构是一样的。

bit_cd,transformer,Transformer,深度学习

 bit_cd,transformer,Transformer,深度学习

注意这个训练完预测的demo是需要加标签的,因为源码默认的读图函数要求输入。

训练效果

bit_cd,transformer,Transformer,深度学习文章来源地址https://www.toymoban.com/news/detail-782524.html

到了这里,关于Github复现-测试基于transformer的变化检测模型BIT_CD的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • 目标检测YOLOv3基于DarkNet53模型测试-笔记

    目标检测YOLOv3基于DarkNet53模型测试-笔记 预测和试测结果: YOLOv3和DarkNet53网络示意图: DarkNet-53网络中Residual Block列差块(基本单元)结构和代码BasicBlock类实现 YOLOv3网络图中基本图的描述 (1)Res Unit即为上图的BasicBlock残差块 (2)CBL即为Conv2D+BatchNorm+LeakReLU组合的卷积基本层 (

    2024年02月14日
    浏览(41)
  • fastspeech2复现github项目--模型构建

    在完成fastspeech论文学习后,对github上一个复现的仓库进行学习,帮助理解算法实现过程中的一些细节;所选择的仓库复现仓库是基于pytorch实现,链接为https://github.com/ming024/FastSpeech2。该仓库是基于https://github.com/xcmyz/FastSpeech中的FastSpeech复现代码完成的,很多代码基本一致。作

    2023年04月21日
    浏览(37)
  • Deepfake检测模型-PyDeepFakeDet项目复现实验报告-深度学习

    实验说明    该实验是对于https://github.com/wangjk666/PyDeepFakeDet模型的复现。 PyDeepFakeDet 是一款集成的、可扩展的 Deepfake 检测工具。目标是提供最先进的 Deepfake 检测模型以及用于在常用 Deepfake 数据集上训练和评估新模型的接口 。 该库 包括基于 CNN 和基于 Transformer 的方法的实现

    2024年02月08日
    浏览(168)
  • 【Transformer】基于Transformer的目标检测算法综述

    本文参考文献:基于Transformer的目标检测算法综述 网络首发时间:2023-01-19 15:01:34 网络首发地址:https://kns.cnki.net/kcms/detail//11.2127.TP.20230118.1724.013.html 在本文中约90%的文字和80%的图片来自该论文,这里只作为学习记录,摘录于此。 深度学习框架Transformer具有强大的建模能力和并

    2023年04月26日
    浏览(58)
  • 目标检测论文解读复现之十:基于YOLOv5的遥感图像目标检测(代码已复现)

    前言        此前出了目标改进算法专栏,但是对于应用于什么场景,需要什么改进方法对应与自己的应用场景有效果,并且多少改进点能发什么水平的文章,为解决大家的困惑,此系列文章旨在给大家解读最新目标检测算法论文,帮助大家解答疑惑。解读的系列文章,本人

    2024年02月06日
    浏览(43)
  • 目标检测论文解读复现【NO.21】基于改进YOLOv7的小目标检测

    前言 此前出了目标改进算法专栏,但是对于应用于什么场景,需要什么改进方法对应与自己的应用场景有效果,并且多少改进点能发什么水平的文章,为解决大家的困惑,此系列文章旨在给大家解读最新目标检测算法论文,帮助大家解答疑惑。解读的系列文章,本人已进行创

    2023年04月09日
    浏览(49)
  • 目标检测论文解读复现之十六:基于改进YOLOv5的小目标检测算法

    前言 此前出了目标改进算法专栏,但是对于应用于什么场景,需要什么改进方法对应与自己的应用场景有效果,并且多少改进点能发什么水平的文章,为解决大家的困惑,此系列文章旨在给大家解读最新目标检测算法论文,帮助大家解答疑惑。解读的系列文章,本人已进行创

    2024年02月03日
    浏览(54)
  • 【论文阅读】基于深度学习的时序异常检测——Anomaly Transformer

    系列文章链接 数据解读参考:数据基础:多维时序数据集简介 论文一:2022 Anomaly Transformer:异常分数预测 论文二:2022 TransAD:异常分数预测 论文三:2023 TimesNet:基于卷积的多任务模型 论文链接:Anomaly Transformer.pdf 代码链接:https://github.com/thuml/Anomaly-Transformer 视频讲解(原

    2024年02月14日
    浏览(37)
  • 基于 Transformation-Equivariant 的自动驾驶 3D 目标检测

    论文地址:https://arxiv.org/abs/2211.11962 论文代码:https://github.com/hailanyi/TED 三维场景中的物体分布有不同的方向。普通探测器不明确地模拟旋转和反射变换的变化。需要大的网络和广泛的数据增强来进行鲁棒检测。 equivariant networks 通过在多个变换点云上应用共享网络显式地模拟

    2024年02月09日
    浏览(43)
  • 基于Python3.7实现的双时相遥感影像变化检测(PaddlePaddle框架和AI Studio平台)

    目 录 第一章 绪论 1 1.1问题背景及意义 1 1.2国内外研究现状 1 1.3论文主要研究内容 2 1.4论文组织结构 3 第二章 相关技术 5 2.1遥感图像变化检测 5 2.2 孪生网络 7 2.3 残差网络 8 2.4 注意力机制 9 第三章 基于BIT的遥感影像变化检测 11 3.1 Transformer模型 11 3.2 BIT模型 12 3.3 AC-BIT模型 1

    2024年01月19日
    浏览(69)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包