TRAIN WITH CUSTOMIZED DATASETS

Published by onesixx on

how to inference, test, and train predefined models with customized datasets.

https://mmdetection.readthedocs.io/en/v2.21.0/2_new_data_model.html#

Train a detector on customized dataset

DataSet 과 Config간의 상호작용

CustomDataset를 상속받아 MyDataSet을 만들고
변환(load_annotations)하여 등록(@DATASETS.register_module)해 놓으면,
mmdetection 프레임웤이 알아서 호출한다. => CustomDataset객체 생성
호출할때 Config로부터 Argument(data_root, ann_file, img_prefix)가 입력된다.

 

customized dataset 준비

config 준비

Train (a new model on the customized dataset)

Test and inference models

Test model

inference models

Categories: vision

onesixx

Blog Owner

Subscribe
Notify of
guest

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