동방프로젝트

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하라고 함.