conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
pip install opencv-python -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.8.0/index.html
pip install terminaltables
pip install pycocotools
python AOI_select.py --AOI_path AOI-data-path
python AOI_get_box --AOI_path AOI-data-path --coco_path COCO_format_path --classes_file class_file
python AOI_to_coco.py --root_dir COCO_format_path --save_path json_file(./train.json)
configs/AD_detection/AD_dsxw_test66.py
python tools/train.py configs/AD_detection/AD_dsxw_test66.py --gpus 1
tools/dist_train.sh configs/AD_detection/AD_dsxw_test66.py 8(GPU_number)
python tools/test.py config_file ckpt_file --eval bbox
python select_threshold.py --config_file config_file --checkpoint_file ckpt_file --images_path testset_path(Contains two folders, OK and ng) --test_batch_size batch_size
python get_score_csv.py --config_file config_file --checkpoint_file ckpt_file --images_path testset_path(unlabel_data) --test_batch_size batch_size --result_path test.csv(Absolute path)