mmclassification Inference a model

Published by onesixx on

Image demo

This script performs inference on a single image.

python demo/image_demo.py \
    ${IMAGE_FILE} \
    ${CONFIG_FILE} \
    ${CHECKPOINT_FILE} \
    [--device ${GPU_ID}] \
    [--score-thr ${SCORE_THR}]

https://mmdetection.readthedocs.io/en/latest/1_exist_data_model.html#inference-and-train-with-existing-models-and-standard-datasets

coco_2017_train데이터셋으로 학습하고, coco_2017_val로 테스트한 모든 모델들은 Model zoo (https://modelzoo.co/)에서 관리

 Faster RCNN (기존 모델)을 활용한다.

=> 즉 기존모델을 활요한다는 것은

해당 Model의 weight를 가지고 있는 checkpoint파일 (https://download.openmmlab.com/mmdetection/v2.0/faster_rcnn/faster_rcnn_r50_fpn_1x_coco/faster_rcnn_r50_fpn_1x_coco_20200130-047c8118.pth)내려 (mmdection/checkpoints/)에 받아서
해당 config파일(mmdetection/configs/faster_rcnn/faster_rcnn_r50_fpn_1x_coco.py)통해,
coco 스타일의 데이터셋으로

Python

Inference a model

Python

Fine-tune(reTrain) a model

ImageNet으로 학습되었던 모델을
내dataset으로 재학습 시킨다.

(작은 데이터로 인한 over-fitting을 방지)

Python
Categories: vision

onesixx

Blog Owner

guest

0 Comments
Inline Feedbacks
View all comments
0
Would love your thoughts, please comment.x
()
x