/MICCAI2022-EMARS

[MICCAI2022] Evolutionary Multi-objective Architecture Search Framework: Application to COVID-19 3D CT Classification

Primary LanguagePythonMIT LicenseMIT

He Xin1, Guohao Ying2, Jiyong Zhang3, and Xiaowen Chu14
1 Hong Kong Baptist University, Hong Kong, China
2 University of Southern California, CA, USA
3 School of Automation, Hangzhou Dianzi University, Hang Zhou, China
4 The Hong Kong University of Science and Technology (Guangzhou), China

Install

pip install -r requirements.txt

search

You can refer to scripts/search_ct.sh for more run scripts.

CUDA_VISIBLE_DEVICES=0 python search.py --config_file ./configs/search.yaml logger.name MyExp

retrain

You can refer to scripts/retrain_ct.sh

there are two mode for retraininigļ¼š

  • you can manually choose a promising architecture by specifying --arch_path to the path of json file, e.g., output/MyExp/version_0/epoch_66.json, and then run the following command
CUDA_VISIBLE_DEVICES=0 python retrain.py --config_file ./configs/retrain.yaml --arc_path outputs/MyExp/version_0/epoch_66.json input.size [128,128]
  • the second is to finetune each selected candidate architecture for a few epochs, and then choose the best-performing one for further training. In this case, you can specify --arc_path to the log path, e.g., output/MyExp/version_0. The json files in this path will be loaded automatically:
CUDA_VISIBLE_DEVICES=0 python retrain.py --config_file ./configs/retrain.yaml --arc_path outputs/MyExp/version_0  input.size [128,128]