동방프로젝트
Iter-E2EDET/SASNet
두원공대88학번뚜뚜
2022. 11. 22. 12:00
SASNet
Iter-E2EDET
Issue에서, 단일 이미지로는 테스트 불가하단 답변 받았다
Sorry to tell you there is no script for the single image testing. However, a possible solution is that you need to compile as *.json like val.json with dummy annotations for all the image testing.
더미 어노테이션을 만들라고 한다.
export NCCL_IB_DISABLE=1
rm -rf __pycache__
rm -rf output/events*
rm -rf output/log.txt.*
python3 train_net.py --num-gpus 4 \
--config-file configs/50e.6h.500pro.ignore.yaml \
--eval-only \
MODEL.WEIGHTS output/model_0039999.pth
이건 run_test.sh 생김새임.
그리고 eval_only가 있다면 진행하는 절차가 있음.
if args.eval_only:
model = Trainer.build_model(cfg)
DetectionCheckpointer(model, save_dir=cfg.OUTPUT_DIR).resume_or_load(cfg.MODEL.WEIGHTS, resume=args.resume)
res = Trainer.test(cfg, model)
if comm.is_main_process():
verify_results(cfg, res)
return res
1) 모델 빌드 후
2) DetectionCheckpointer 진행 후
3) Trainer.test 하라고 한다.
4) comm이 main process라면 verify_result하라고 함.