This repository contains the code I wrote for the 6th digix global ai challenge:BI_RADS. Final Ranking: 23rd out of 240. Now I am open-sourcing the code, hoping to provide relevant assistance to those working in this field.
Before running the code, ensure that you have a suitable Python environment with necessary dependencies installed. You can install the dependencies via:
pip install -r requirements.txt
If you want to conduct training, I recommend using publicly available open-source datasets.
To train feature model, use the following command:
python train_fea.py
To train classification model, use the following command:
python train_cla.py
Note:We use four expert models to achieve feature detection.
After training the model and using the correct model address, input the following code to make predictions.
python run.py