Tong Zhang 614ffd9d23 | 1 year ago | |
---|---|---|
data | 3 years ago | |
experiments/miccai 2021 | 2 years ago | |
images | 3 years ago | |
models | 3 years ago | |
readme | 1 year ago | |
scripts | 2 years ago | |
src | 2 years ago | |
.gitignore | 3 years ago | |
LICENSE | 3 years ago | |
NOTICE | 3 years ago | |
README.md | 1 year ago | |
requirements.txt | 2 years ago | |
tags | 2 years ago |
by Jiancong Chen, Yingying Zhang, Jingyi Wang, Xiaoxue Zhou, Yihua He*, Tong Zhang*.
This repo contains the official pytorch implemetation for EllipseNet.
Please refer to INSTALL.md for installation instructions.
Methods | Setting | DiceT | DiceC | Diceall | Pavg |
---|---|---|---|---|---|
EllipseNet (exp6) | only IoU loss | 0.8813 | 0.8520 | 0.8666 | 0.8855 |
EllipseNet (exp1) | w/o IoU loss | 0.9338 | 0.9108 | 0.9224 | 0.8841 |
EllipseNet (exp3) | w/ IoU loss | 0.9430 | 0.9224 | 0.9336 | 0.8949 |
Prepare the elliptical dataset in coco-format. An example script is given in scripts/prepare_label.ipynb. We provide scripts for all the experiments in the experiments folder.
Usage:
chmod +x experiments/miccai21/*.sh
./experiments/miccai21/exp3_base_theta5_iou.sh
If you need the docker for reproduction, please contact via email. We will provide the docker image.
EllipseNet itself is released under the MIT License (refer to the LICENSE file for details).
Portions of the code are borrowed from CenterNet, Rotated_IoU, human-pose-estimation.pytorch (image transform, resnet), CornerNet (hourglassnet, loss functions), dla (DLA network), DCNv2 (deformable convolutions), tf-faster-rcnn (Pascal VOC evaluation) and kitti_eval (KITTI dataset evaluation). Please refer to the original License of these projects (See NOTICE).
If you find this project useful for your research, please use the following BibTeX entry.
Chen J., Zhang Y., Wang J., Zhou X., He Y., Zhang T*. (2021) EllipseNet: Anchor-Free Ellipse Detection for Automatic Cardiac Biometrics in Fetal Echocardiography. In: de Bruijne M. et al. (eds) Medical Image Computing and Computer Assisted Intervention – MICCAI 2021. MICCAI 2021. Lecture Notes in Computer Science, vol 12907. Springer, Cham. https://doi.org/10.1007/978-3-030-87234-2_21
If you have any questions about this paper, welcome to email to zhangt02@pcl.ac.cn
No Description
Jupyter Notebook Python C++ C Cuda other
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》