Tong Zhang 59be300120 | 1 year ago | |
---|---|---|
data/gan | 2 years ago | |
imgs | 2 years ago | |
networks | 3 years ago | |
u_d | 3 years ago | |
utils | 3 years ago | |
.gitignore | 2 years ago | |
LICENSE | 3 years ago | |
README.md | 1 year ago | |
identical_mapping.py | 3 years ago | |
predict.py | 3 years ago | |
requiresment.txt | 2 years ago | |
train.py | 2 years ago |
by Yang Yang, Jiancong Chen, Ruixuan Wang*, Ting Ma, Lingwei Wang, Jie Chen, Wei-Shi Zheng, Tong Zhang*.
@inproceedings{yang2021towards,
title={Towards Unbiased COVID-19 Lesion Localisation and Segmentation via Weakly Supervised Learning},
author={Yang, Yang and Chen, Jiancong and Wang, Ruixuan and Ma, Ting and Wang, Lingwei and Chen, Jie and Zheng, Wei-Shi and Zhang, Tong},
booktitle={IEEE International Symposium on Biomedical Imaging},
pages={1966--1970},
year={2021},
organization={IEEE}
}
Despite tremendous efforts, it is very challenging to generate a robust model to assist in the accurate quantification assessment of COVID-19 on chest CT images. Due to the nature of blurred boundaries, the supervised segmentation methods usually suffer from annotation biases. To support unbiased lesion localisation and to minimise the labelling costs, we propose a data-driven framework supervised by only image level labels. The framework can explicitly separate potential lesions from original images, with the help of an generative adversarial network and a lesion-specific decoder. Experiments on two COVID-19 datasets demonstrates the effectiveness of the proposed framework and its superior performance to several existing methods.
Python 3.8+
Pytorch 1.10+
Torchvision 0.11.0
GPU, such as a 1080Ti
You can install require modules with following command.
pip install -r requirements.txt
Data is stored in ./data, you can directory download it (~400M).
You can train with following code, which save logs and models to /exp/exp-1.
python train.py -p ./exp/exp-1 -l 10 -s 0.01 -a 0.01 -g 200 -t 0 -i 15 -b 4 -e 120 --debug
If you find this project useful for your research, please cit our work use the following BibTeX entry.
@INPROCEEDINGS{9433806,
author={Yang, Yang and Chen, Jiancong and Wang, Ruixuan and Ma, Ting and Wang, Lingwei and Chen, Jie and Zheng, Wei-Shi and Zhang, Tong},
booktitle={2021 IEEE 18th International Symposium on Biomedical Imaging (ISBI)},
title={Towards Unbiased Covid-19 Lesion Localisation And Segmentation Via Weakly Supervised Learning},
year={2021},
volume={},
number={},
pages={1966-1970},
doi={10.1109/ISBI48211.2021.9433806}}
If you have any questions about this paper, welcome to email to zhangt02@pcl.ac.cn
Dear OpenI User
Thank you for your continuous support to the Openl Qizhi Community AI Collaboration Platform. In order to protect your usage rights and ensure network security, we updated the Openl Qizhi Community AI Collaboration Platform Usage Agreement in January 2024. The updated agreement specifies that users are prohibited from using intranet penetration tools. After you click "Agree and continue", you can continue to use our services. Thank you for your cooperation and understanding.
For more agreement content, please refer to the《Openl Qizhi Community AI Collaboration Platform Usage Agreement》