YOLOv4-MultiTask
YOLOv4-MultiTask copied to clipboard
YOLOv4-MultiTask for Pytorch
训练数据增强
@Huangdebo 请教下,大佬在训练时候,做了哪些数据增强的啊,怎么保持车道线label与数据增强之后的图像一致呢
训练部分
感谢分享了很好的方案。对于训练部分很感兴趣,如果有机会想向您请教训练部分的问题。
hi,非常感谢分享pre-trained model。 pytorch model可以正常检测图像 但是caffemodel部分:detection_out的分支可以解析出目标,但lane分支按照detect_video.py 以下这段code,结果始终有问题: out_j = lane[0].data.cpu().numpy() out_j = out_j[:, ::-1, :] prob = scipy.special.softmax(out_j[:-1, :, :], axis=0) idx = np.arange(100) + 1 idx = idx.reshape(-1, 1, 1)...
大佬能说下这个两个任务的训练损失最后是怎么结合的吗,想学习下设计思路
你好,我在运行detect_demo.py时,模型加载错误。请问这个是什么原因呢? Missing key(s) in state_dict: "total_ops", "total_params", "module_list.total_ops", "module_list.total_params", "module_list.0.total_ops", "module_list.0.total_params", "module_list.0.Conv2d.total_ops",............