Hice la siguiente modificacion para cuando se trabaje con otro dataset no tener que estar pendiente de que tan grandes son las imagenes:
y_dim, x_dim, c_dim = img.shape
print(y_dim, x_dim, c_dim)
gt_boxes={}
for index, row in df.iterrows():
if row['class_id'] == 3:
bbox = np.array([[row['ymin']/y_dim, row['xmin']/x_dim, row['ymax']/y_dim, row['xmax']/x_dim]], dtype=np.float32)
if row['frame'] in gt_boxes:
gt_boxes[row['frame']] = np.append(gt_boxes[row['frame']], np.array(bbox), axis=0)
else:
gt_boxes[row['frame']] = np.array(bbox)
¿Quieres ver más aportes, preguntas y respuestas de la comunidad?