This article will be used to record fracture detection. The overall process will be to first detect and then classify the vertebral body in the detection box to determine whether it is fractured or not.
see details
- Upload drr detection weights.
- add yolov5 detection predict.
- add detection training.
Install
pip install -r requirements.txt
Data
.
├── cut_drr
│ ├── AP
│ │ ├── fracture_images
│ │ └── normal_images
│ ├── LA
│ │ ├── fracture_images
│ │ └── normal_images
│ ├── all
│ │ ├── fracture_images
│ │ └── normal_images
Training
[Note] the pretrain model can download from the Original repository.python train.py
Predict
python predict.py
Eval
python eval.py
Detection classify
you can choose run rtdetr_paddle or rtdetr_pytorchpython detection classify.py