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数据文件夹
python AOI_get_box --AOI_path AOI数据文件夹 --coco_path COCO格式文件夹 --classes_file 类别文件
python AOI_to_coco.py --root_dir COCO格式文件夹 --save_path json标签命名(./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数量)
python tools/test.py 参数文件 权重文件 --eval bbox
python select_threshold.py --config_file 模型参数文件 --checkpoint_file 模型权重文件 --images_path 测试集路径(包含ok和ng两个文件夹) --test_batch_size 测试阶段的batch size大小
python get_score_csv.py --config_file 模型参数文件 --checkpoint_file 模型权重文件 --images_path 测试集路径(未标注数据) --test_batch_size 测试阶段的batch size大小 --result_path test.csv(绝对路径)