Retrain YOLOv5 on a custom dataset

@Marina_Vilela_Bento
is it correctly run the hef file without any error