专栏算法工具链bev_lss模型训练求助

bev_lss模型训练求助

已解决
chengang2023-07-21
73
3
1.芯片型号：J5
2.天工开物开发包OpenExplorer版本：J5_OE_1.1.57
3.问题定位：模型训练问题

你们好，我有2个问题需要咨询：

1）训练模型时会出现意外中断，现在每次都要从头重新训练，地平线OE包支持加载训练好的模型继续训练吗，这部分需要修改哪里的代码来实现？

2）关于bev_lss模型生成参考点，homo矩阵转换到图像像素坐标时（X,Y,D），会将不在特征图范围内的点判为invalid点，这里为什么是根据特征图大小来判断(16，44)，而不是根据原图img_size来判断（256,704），请帮忙解答，谢谢。

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2

    你好，对于问题1，地平线目前的训练架构支持恢复意外中断的训练，只需要在config文件的trainer中配置resume_optimizer和resume_epoch_or_step参数，详细的配置说明请参考用户手册【6.1.3.1.3 恢复训练】章节https://developer.horizon.cc/api/v1/fileData/horizon_j5_open_explorer_cn_doc/hat/source/tutorials/quantization.html#id5

    问题2：首先，view_transformer.py 中的_get_homography函数里，已经把homo矩阵scale到了图像特征的尺度，所以当使用homo矩阵把上图中的coords转换到new_coords(图像特征采样点)时，是把图像特征大小以外的点去除，而不是根据原图img_size来判断。

    2023-07-21
    0
    2
    • chengang回复颜值即正义:

      非常感谢，很有帮助。

      2023-07-21
      0
    • 颜值即正义回复chengang:

      您好，地平线工具链在持续迭代优化，为了给您提供更好的服务，希望您能抽出3分钟左右的时间，将您在使用工具链期间的感受和建议告诉我们，您的宝贵意见对我们很重要，非常感谢！https://wenjuan.feishu.cn/m/?t=st64p6krU3Ji-yvhv

      2023-07-21
      0