qqwweee / keras-yolo3

A Keras implementation of YOLOv3 (Tensorflow backend)
MIT License
7.14k stars 3.44k forks source link

Train Single Label Detection #749

Closed malik-anhar closed 3 years ago

malik-anhar commented 3 years ago

Let's say I wanna train the model only for single label (face), so what is the correct annotation for the train.txt? Is it something like this

path/to/img1.jpg 50,100,150,200,0 30,50,200,120,0
path/to/img2.jpg 120,300,250,600,0

or something like this

path/to/img1.jpg 50,100,150,200,1 30,50,200,120,1
path/to/img2.jpg 120,300,250,600,1

The class_id should be 0 or 1?

Anyman552 commented 3 years ago

The class_id should be 0