yolov5 face 训练自己的数据集优化笔记
价值上万,训练效果很好,
目录
main函数启动参数:
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--weights', type=str, default='yolov5-blazeface.pt', help='initial weights path')
parser.add_argument('--cfg', type=str, default='models/blazeface.yaml', help='model.yaml path')
parser.add_argument('--data', type=str, default='data/widerface_lbg.yaml', help='data.yaml path')
parser.add_argument('--hyp', type=str,