taformer
Science Score: 44.0%
This score indicates how likely this project is to be science-related based on various indicators:
-
✓CITATION.cff file
Found CITATION.cff file -
✓codemeta.json file
Found codemeta.json file -
✓.zenodo.json file
Found .zenodo.json file -
○DOI references
-
○Academic publication links
-
○Academic email domains
-
○Institutional organization owner
-
○JOSS paper metadata
-
○Scientific vocabulary similarity
Low similarity (8.8%) to scientific vocabulary
Repository
Basic Info
- Host: GitHub
- Owner: CVIU-CSU
- License: apache-2.0
- Language: Python
- Default Branch: master
- Size: 4.65 MB
Statistics
- Stars: 0
- Watchers: 2
- Forks: 0
- Open Issues: 0
- Releases: 0
Metadata Files
README.md
Task-Aware Transformer For Partially Supervised Retinal Fundus Image Segmentation
Environment
This code is based on mmsegmentation
python == 3.8
Pytorch == 1.9.0
timm
imgviz
```shell conda create --name mmseg-v1rc python=3.8 -y conda activate mmseg-v1rc
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=10.2 -c pytorch -y
pip install -U openmim mim install "mmengine==0.3.2" mim install "mmcv==2.0.0rc3"
git clone https://github.com/open-mmlab/mmsegmentation.git cd mmsegmentation git checkout -b 1rc2 v1.0.0rc2 pip install -v -e . -i https://pypi.douban.com/simple/
mim install "mmdet==3.0.0rc4" mim install "mmcls==1.0.0rc4"
pip install imgviz pip install timm pip install kornia==0.5.8 ```
Note that replacing mmengine, mmcv and mmdet in the environment directory lib/python3.8/site-packages with the given packages.
Dataset Preparations
Please see PSSNet/prepare_dataset
Training
shell
bash mmsegmentation/tools/dist_train.sh mmsegmentation/configs/_mask2former_/mask2former_swin-t_4xb2-40k_multi_dataset-512x512_90query_6layer_mean_teacher_query_mask_random_shift_learn_generate_query_mask_feature_pseudo_factor.py 4
Evaluation
shell
python mmsegmentation/tools/vis.py images_path ann_path 7 config_file_path checkpoint_path
Models
We provide the final model and training logs here
Citation
If you find it useful for your your research and applications, please cite using this BibTeX:
bibtex
@inproceedings{zeng2024task,
title={Task-Aware Transformer For Partially Supervised Retinal Fundus Image Segmentation},
author={Zeng, Hailong and Liu, Jianfeng and Liang, Yixiong},
booktitle={2024 International Joint Conference on Neural Networks (IJCNN)},
pages={1--8},
year={2024},
organization={IEEE}
}
Owner
- Name: CVIU-CSU
- Login: CVIU-CSU
- Kind: organization
- Repositories: 4
- Profile: https://github.com/CVIU-CSU
Citation (CITATION.cff)
cff-version: 1.2.0 message: "If you use this software, please cite it as below." authors: - name: "MMSegmentation Contributors" title: "OpenMMLab Semantic Segmentation Toolbox and Benchmark" date-released: 2020-07-10 url: "https://github.com/open-mmlab/mmsegmentation" license: Apache-2.0
GitHub Events
Total
Last Year
Dependencies
- pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
- pytorch/pytorch ${PYTORCH}-cuda${CUDA}-cudnn${CUDNN}-devel build
- docutils ==0.16.0
- myst-parser *
- sphinx ==4.0.2
- sphinx_copybutton *
- sphinx_markdown_tables *
- mmcls >=1.0.0rc0
- mmcv >=2.0.0rc3,<2.1.0
- mmdet >=3.0.0rc4
- mmengine >=0.1.0,<1.0.0
- cityscapesscripts *
- nibabel *
- mmcv >=2.0.0rc0
- mmengine *
- prettytable *
- torch *
- torchvision *
- matplotlib *
- numpy *
- packaging *
- prettytable *
- scipy *
- codecov * test
- flake8 * test
- interrogate * test
- pytest * test
- xdoctest >=0.10.0 test
- yapf * test