catnet
🛰️ Learning to Aggregate Multi-Scale Context for Instance Segmentation in Remote Sensing Images (TNNLS 2024)
Science Score: 36.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
○CITATION.cff file
-
✓codemeta.json file
Found codemeta.json file -
○.zenodo.json file
-
✓DOI references
Found 4 DOI reference(s) in README -
✓Academic publication links
Links to: arxiv.org, researchgate.net, ieee.org -
○Committers with academic emails
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (12.8%) to scientific vocabulary
Keywords
Repository
🛰️ Learning to Aggregate Multi-Scale Context for Instance Segmentation in Remote Sensing Images (TNNLS 2024)
Basic Info
- Host: GitHub
- Owner: yeliudev
- License: gpl-3.0
- Language: Python
- Default Branch: main
- Homepage: https://arxiv.org/abs/2111.11057
- Size: 129 KB
Statistics
- Stars: 82
- Watchers: 1
- Forks: 12
- Open Issues: 0
- Releases: 0
Topics
Metadata Files
README.md
Context Aggregation Network
This repository maintains the official implementation of the paper Learning to Aggregate Multi-Scale Context for Instance Segmentation in Remote Sensing Images by Ye Liu, Huifang Li, Chao Hu, Shuang Luo, Yan Luo, and Chang Wen Chen, which has been accepted by TNNLS.
Installation
Please refer to the following environmental settings that we use. You may install these packages by yourself if you meet any problem during automatic installation.
- CUDA 11.8
- CUDNN 8.7.0.84
- Python 3.11.3
- PyTorch 2.0.1
- MMEngine 0.7.4
- MMCV 2.0.0
- MMDetection 3.0.0
- NNCore 0.3.6
Install from source
- Clone the repository from GitHub.
git clone https://github.com/yeliudev/CATNet.git
cd CATNet
- Install dependencies.
pip install -r requirements.txt
- Set environment variable
export PYTHONPATH=$PWD:$PYTHONPATH
Getting Started
Download and prepare the datasets
- Download the pre-processed datasets.
- Prepare the files in the following structure.
CATNet
├── configs
├── datasets
├── models
├── tools
├── data
│ ├── dior
│ │ ├── Annotations
│ │ ├── ImageSets
│ │ └── JPEGImages
│ ├── hrsid
│ │ ├── annotations
│ │ └── images
│ ├── isaid
│ │ ├── train
│ │ ├── val
│ │ └── test
│ └── vhr
│ ├── annotations
│ └── images
├── README.md
├── setup.cfg
└── ···
Train a model
Run the following command to train a model using a specified config.
mim train mmdet <path-to-config> --gpus 4 --launcher pytorch
If an
out-of-memoryerror occurs on iSAID dataset, please uncomment L22-L24 in the dataset code and try again. This will filter out a few images with more than 1,000 objects, largely reducing the memory cost.
Test a model and evaluate results
Run the following command to test a model and evaluate results.
mim test mmdet <path-to-config> --checkpoint <path-to-checkpoint> --gpus 4 --launcher pytorch
Model Zoo
We provide multiple pre-trained models here. All the models are trained using 4 NVIDIA A100 GPUs and are evaluated using the default metrics of the datasets.
| Dataset | Model | Backbone | Schd | Aug | Performance | Download | |
|---|---|---|---|---|---|---|---|
| BBox AP | Mask AP | ||||||
| iSAID | CAT Mask R-CNN | ResNet-50 | 3x | ✗ | 45.1 | 37.2 | model | metrics |
| CAT Mask R-CNN | ResNet-50 | 3x | ✓ | 47.7 | 39.2 | model | metrics | |
| DIOR | CATNet | ResNet-50 | 3x | ✗ | 74.0 | — | model | metrics |
| CATNet | ResNet-50 | 3x | ✓ | 78.2 | — | model | metrics | |
| CAT R-CNN | ResNet-50 | 3x | ✗ | 75.8 | — | model | metrics | |
| CAT R-CNN | ResNet-50 | 3x | ✓ | 80.6 | — | model | metrics | |
|
NWPU VHR-10 |
CAT Mask R-CNN | ResNet-50 | 6x | ✗ | 71.0 | 69.3 | model | metrics |
| CAT Mask R-CNN | ResNet-50 | 6x | ✓ | 72.4 | 70.7 | model | metrics | |
| HRSID | CAT Mask R-CNN | ResNet-50 | 6x | ✗ | 70.9 | 57.6 | model | metrics |
| CAT Mask R-CNN | ResNet-50 | 6x | ✓ | 72.0 | 59.6 | model | metrics | |
Citation
If you find this project useful for your research, please kindly cite our paper.
bibtex
@article{liu2024learning,
title={Learning to Aggregate Multi-Scale Context for Instance Segmentation in Remote Sensing Images},
author={Liu, Ye and Li, Huifang and Hu, Chao and Luo, Shuang and Luo, Yan and Chen, Chang Wen},
journal={IEEE Transactions on Neural Networks and Learning Systems},
year={2024}
}
Owner
- Name: Ye Liu
- Login: yeliudev
- Kind: user
- Location: Hong Kong SAR, China
- Company: The Hong Kong Polytechnic University
- Website: https://yeliu.dev/
- Repositories: 9
- Profile: https://github.com/yeliudev
GitHub Events
Total
- Issues event: 1
- Watch event: 18
- Issue comment event: 1
- Fork event: 6
Last Year
- Issues event: 1
- Watch event: 18
- Issue comment event: 1
- Fork event: 6
Issues and Pull Requests
Last synced: 7 months ago
All Time
- Total issues: 17
- Total pull requests: 0
- Average time to close issues: 13 days
- Average time to close pull requests: N/A
- Total issue authors: 13
- Total pull request authors: 0
- Average comments per issue: 1.71
- Average comments per pull request: 0
- Merged pull requests: 0
- Bot issues: 0
- Bot pull requests: 0
Past Year
- Issues: 5
- Pull requests: 0
- Average time to close issues: 13 days
- Average time to close pull requests: N/A
- Issue authors: 4
- Pull request authors: 0
- Average comments per issue: 2.6
- Average comments per pull request: 0
- Merged pull requests: 0
- Bot issues: 0
- Bot pull requests: 0
Top Authors
Issue Authors
- wangqingl (2)
- selinkoles (2)
- gsj-zk (2)
- wuyunxiangwyx (1)
- 348300984 (1)
- lqh964165950 (1)
- vansin (1)
- chhhhh123 (1)
- rose-jinyang (1)
- YangPanHZAU (1)
- talhayaseen57 (1)
- cashily (1)
Pull Request Authors
Top Labels
Issue Labels
Pull Request Labels
Dependencies
- mmcv-full >=1.3.14,<1.4
- mmdet >=2.18,<2.19
- nncore >=0.3.2
- torch >=1.10
- torchvision >=0.11
- actions/checkout v3 composite
- actions/setup-python v4 composite