search

yjh0410 / YOWOv2

The second generation of YOWO action detector.
MIT License
187 stars 31 forks source link

train.py #14

Open ChethanN01 opened 1 year ago

ChethanN01 commented 1 year ago

https://github.com/yjh0410/YOWOv2/blob/master/train.py Where is the class name feed into model, i havent see class name have passed while training!

Batman-97 commented 1 year ago

Hello @ChethanN01,

I hope you were asking about this config file where you can change the class name according to your dataset.

dataset_config = {