Обучение Yolo требует следующего формата аннотации
[class] [X] [Y] [W] [H]
т.е.
0 0.443359 0.434722 0.391406 0.869444
Если у вас такой же набор данных аннотации, поздравляем васмогу начать тренировку.Если у вас его нет, вы можете конвертировать, используя инструменты, доступные на github.
Обновление: Аннотация рассчитывается по центру Например.Если вы работаете с форматом x1, y1, x2, y2, вам необходимо конвертировать его.
def convert_to_yolo_format(path, x1, y1, x2, y2):
"""
Definition: Converts (x1, y1, x1, y2) format to
(x, y, width, height) normalized YOLO format.
"""
size = get_img_shape(path) # size array [width,height]
dw = 1./size[0]
dh = 1./size[1]
x = (x1 + x2)/2.0 # centroid x
y = (y1 + y2)/2.0 # centroid y
w = x2 - x1 # width
h = y1 - y2 # height
x = abs(x*dw) # divide by width
w = abs(w*dw) # divide by width
y = abs(y*dh) # divide by height
h = abs(h*dh) # divide by height
return (x,y,w,h)