search

j-marple-dev / AYolov2

GNU General Public License v3.0
156 stars 15 forks source link

Custom dataset training #57

Closed cosao closed 2 years ago

cosao commented 2 years ago

image The class does not change during model training using custom data. Please tell me how to solve this problem.

JeiKeiLim commented 2 years ago

The label names were supposed to be defined in data.yaml but we didn't have time to implement it.

For now label names are defined as scripts.utils.constants.LABELS It's temporary but you can add your own dataset label in the dictionary.

LABELS = {
   "COCO": [ ... ],
   "VOC": [ ... ],
   "CUSTOM_DATASET": [ "label_1", "label_2", ... ]
}

And pass dataset_name argument in LoadImagesAndLabels class.

cosao commented 2 years ago

i solved problem, thanks for your answer