Ecosyste.ms: Awesome
An open API service indexing awesome lists of open source software.
https://github.com/AndreaCodegoni/Tiny_model_4_CD
Official implementation of TinyCD: A (Not So) Deep Learning Model For Change Detection
https://github.com/AndreaCodegoni/Tiny_model_4_CD
Last synced: 3 months ago
JSON representation
Official implementation of TinyCD: A (Not So) Deep Learning Model For Change Detection
- Host: GitHub
- URL: https://github.com/AndreaCodegoni/Tiny_model_4_CD
- Owner: AndreaCodegoni
- Created: 2022-06-15T08:55:31.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2024-05-09T09:34:10.000Z (6 months ago)
- Last Synced: 2024-07-23T03:38:49.715Z (4 months ago)
- Language: Python
- Homepage:
- Size: 6.76 MB
- Stars: 111
- Watchers: 2
- Forks: 19
- Open Issues: 3
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
- awesome-remote-sensing-change-detection - Codegoni A, Lombardi G, Ferrari A. TINYCD: A (Not So) Deep Learning Model For Change Detection
README
# Tiny_CD
This is the implementation of: TinyCD: A (Not So) Deep Learning Model For Change Detection
:fire: :fire: :fire: TinyCD has been accepted for publication in [Neural Computing and Applications](https://link.springer.com/article/10.1007/s00521-022-08122-3) :fire: :fire: :fire:
You can find the pre-print version here: [arXiv](https://arxiv.org/abs/2207.13159)
## Results
https://user-images.githubusercontent.com/54067932/180251485-b3776855-3666-460a-a15b-05b2bbc0a088.mp4
In the following table we report the quantitative comparison of our model and other State-of-the-art models. F1 scores are reported in percentage, parameters in Millions and Flops in Gigaflops
Here we report a visual comparison between the mask generated by TinyCD and BIT.
We selected negative samples in the fairness way in order to show both similarities and differences between the obtained results.
Finally, we show a complete sequence of intermediate masks and final binary classification mask to show the ability of our tiny model in detecting detailed objects.
## Installation and Requirements
The easiest way to reproduce our results is to create a folder named "TinyCD" on your device and then
clone this repository in "TinyCD":
```shell
git clone https://github.com/AndreaCodegoni/Tiny_model_4_CD.git
```
Then, you can create a virtual ``conda`` environment named ``TinyCD`` with the following cmd:
```shell
conda create --name TinyCD --file requirements.txt
conda activate TinyCD
```
## Dataset
You can find the original datasets at these two links:
LEVIR-CD: https://justchenhao.github.io/LEVIR/
WHU-CD: https://study.rsgis.whu.edu.cn/pages/download/building_dataset.html
Then, for each dataset, you have to organise the data in the following way:
`A`: images of t1 phase;
`B`: images of t2 phase;
`label`: label maps;
`list`: contains `train.txt`, `val.txt` and `test.txt`, each file records the image names (XXX.png) in the change detection dataset.
If you prefer, you can download the pre-processed dataset from this [link](https://drive.google.com/drive/folders/1uTExi7JbiSwXXqPdeJp2GCE7B-0QI6xh?usp=sharing).
If you have any trouble with the datasets, feel free to contact us.
## Evaluate pretrained models
If you want to evaluate your trained model, or if you want to reproduce the paper results with the pretrained models that
you can find in the "pretrained_models" folder, you can run:
```cmd
python test_ondata.py --datapath "Your_data_path" --modelpath "Your_path_to_pretrained_model"
```
## Train your model
You can re-train our model, or if you prefer you can play with the parameters of our model and then train it using
```cmd
python training.py --datapath "Your_data_path" --log-path "Path_to_save_logs_and_models_checkpoints"
```
## External repositories using TinyCD
Here I report some repositories that use TinyCD for their search.
If you use TinyCD and enjoy it, please let me know your work so I can add it
- [Exstension on images with >3 channels (e.g. sentinel 2)](https://github.com/Dibz15/OpenMineChangeDetection) (thanks to @robmarkcole for the link)
- [Open-CD is an open source change detection toolbox based on a series of open source general vision task tools.](https://github.com/likyoo/open-cd)
- [A Machine Learning data pipeline for the Change detection for Burned area Delineation (ChaBuD) challenge at the ECML/PKDD 2023 conference.](https://github.com/developmentseed/chabud2023/tree/main) (thanks to @robmarkcole for the link)
## References
We want to mention the following repositories that greatly help us in our work:
- https://github.com/justchenhao/BIT_CD We have used this repository in the visual comparison and to report other state-of-the-art results on the two datasets.
- https://github.com/wgcban/ChangeFormer/ and https://github.com/wgcban/SemiCD for the datasets.
## License
Code is released for non-commercial and research purposes **only**. For commercial purposes, please contact the authors.