Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.
Tong Zhang e27e2ef90c | 1 year ago | |
---|---|---|
ascend310_infer | 1 year ago | |
infer | 1 year ago | |
kernel_meta | 1 year ago | |
modelarts | 1 year ago | |
ms_log | 1 year ago | |
scripts | 1 year ago | |
somas_meta | 1 year ago | |
src | 1 year ago | |
README.md | 1 year ago | |
create_dataset.log | 1 year ago | |
default_config.yaml | 1 year ago | |
default_config.yaml.backup | 1 year ago | |
eval.py | 1 year ago | |
eval_log.txt | 1 year ago | |
export.py | 1 year ago | |
mindspore_hub_conf.py | 1 year ago | |
postprocess.py | 1 year ago | |
preprocess.py | 1 year ago | |
requirements.txt | 1 year ago | |
tags | 1 year ago | |
train.py | 1 year ago | |
train_log.txt | 1 year ago | |
training_log.txt | 1 year ago |
We use CenterNet as backbone to detect lung nodules. CenterNet is a novel practical anchor-free method for object detection, 3D detection, and pose estimation, which detect identifies objects as axis-aligned boxes in an image. The detector uses keypoint estimation to find center points and regresses to all other object properties, such as size, 3D location, orientation, and even pose. In nature, it's a one-stage method to simultaneously predict center location and bboxes with real-time speed and higher accuracy than corresponding bounding box based detectors.
Paper Setio AA, Traverso A, De Bel T, Berens MS, Van Den Bogaard C, Cerello P, Chen H, Dou Q, Fantacci ME, Geurts B, van der Gugten R. Validation, comparison, and combination of algorithms for automatic detection of pulmonary nodules in computed tomography images: the LUNA16 challenge. Medical image analysis. 2017 Dec 1;42:1-3.
The stacked Hourglass Network downsamples the input by 4×,followed by two sequential hourglass modules.Each hourglass module is a symmetric 5-layer down-and up-convolutional network with skip connections .This network is quite large ,but generally yields the best keypoint estimation performance.
Note that you can run the scripts based on the dataset mentioned in original paper or widely used in relevant domain/network architecture. In the following sections, we will introduce how to run the scripts using the related dataset below.
Dataset used: LUNA16 It consists of 1,186 lung nodules annotated in 888 CT scans.
Dataset size:26G
Data format:image and json files
Note:Data will be processed in dataset.py
Hardware(Ascend)
Framework
pip install -r requirements.txt
Dataset should be organize as COCO format and use convert_dataset_to_mindspore.sh to convert data to mindspore.
PATH in shell may need to update according to your dataset location.
```shell
# create dataset in mindrecord format
bash scripts/convert_dataset_to_mindrecord.sh
# standalone training on Ascend
bash scripts/run_standalone_train_ascend.sh
# eval on Ascend
bash scripts/run_standalone_eval_ascend.sh
```
Inference
#eval_txt.log
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.513
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.938
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.507
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.508
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.593
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.586
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.602
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.602
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.599
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.631
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = -1.000
CenterNet on 11.8K images(The annotation and data format must be the same as coco)
Parameters | CenterNet_Hourglass |
---|---|
Resource | Ascend 910; CPU 2.60GHz, 192cores; Memory, 755G |
uploaded Date | 3/27/2021 (month/day/year) |
MindSpore Version | 1.3.0 |
Dataset | LungDetection |
Training Parameters | 8p, epoch=130, steps=158730, batch_size = 12, lr=2.4e-4 |
Optimizer | Adam |
Loss Function | Focal Loss, L1 Loss, RegLoss |
outputs | detections |
Checkpoint | 2.1G (.ckpt file) |
CenterNet on validation(5K images) and test-dev(40K images)
Parameters | CenterNet_Hourglass |
---|---|
Resource | Ascend 910; CPU 2.60GHz, 192cores; Memory, 755G |
uploaded Date | 11/27/2021 (month/day/year) |
MindSpore Version | 1.1.0 |
Dataset | LungDetection |
batch_size | 1 |
outputs | mAP |
Accuracy(validation) | MAP: 41.5%, AP50: 60.4%, AP75: 44.7%, Medium: 45.7%, Large: 53.6% |
CenterNet on validation(5K images)
Parameters | CenterNet_Hourglass |
---|---|
Resource | Ascend 310; CentOS 3.10 |
uploaded Date | 11/31/2021 (month/day/year) |
MindSpore Version | 1.3.0 |
Dataset | LungDetection |
batch_size | 1 |
outputs | mAP |
Accuracy(validation) | MAP: 51.3% |
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》