Please check INSTALL.md for installation instructions. Please prepare the data following this repo
The following command line controls different stages of training:
# Train the source only model
python tools/train_net_mcd.py --config-file configs/sf/source_only.yaml SOLVER.SFDA_STAGE 1
# Train the model with DUT on Abnormal Symptoms dataset
python tools/train_net_mcd.py --config ./configs/sf/dut_hcmus.yaml OUTPUT_DIR outputs/dut_hcmus
# Train the model with DUT on WCE dataset
python tools/train_net_mcd.py --config ./configs/sf/dut_hcmus.yaml OUTPUT_DIR outputs/dut_hcmusThe code is based on FCOS. For enquiries please contact xliu423-c@my.cityu.edu.hk.