train: train/images
val: val/images
nc: 2
names: ['cat', 'dog']