专栏算法工具链训练自己的数据集（cityscapes格式）

训练自己的数据集（cityscapes格式）

已解决
默认213442024-07-24
58
8

问题：

1. 芯片型号：J5

2. 天工开物开发包 OpenExplorer 版本：J5_OE_1.1.68

3. 问题定位：模型训练

4. 问题具体描述：见附件。

附件:
算法工具链
征程5
评论4
0/1000
  • GOYU\XU
    Lv.2
    训练开源的cityscapes数据集没有问题，训练程序应该没问题。从自有数据训练情况看也没报错，只是精度出问题了

    建议从以下几点排查问题：

    1. 确保自有数据转换后格式同cityscapes格式一致

    2. 检测数据加载处代码是否需要修改的

    3. 检测head处训练代码是否需要修改的

    2024-07-24
    1
    1
    • 默认21344回复GOYU\XU:

      您好，我现在应该可以训练了，但是使用以下命令到处 onnx 模型，进行推理：

      参数：stage==qat， 通过 export_to_onnx(model, (example_input, {}), file_path, **kwargs) 转换。

      通过 04_inference.sh 进行推理，发现获得的图片，与原图片一样，没有mask蒙版。

      在 postprocess.py 中的 plot_image()函数中：

      out_img 是一张全0图片。

      附：魔性推理出来的tensor是整型的，不是浮点型的，并且也是全0的。


      问题：

      1、我应该如何处理模型输出，是参数的问题吗？还是我训练代码存在问题，没有成功训练。

      2、是否有其他文件可以直接用 .pth.tar 参数文件直接推理图片文件夹。

      2024-07-26
      1
  • GOYU\XU
    Lv.2
    模型转onnx为什么选择stage==qat呀，可能是模型转onnx这步出问题了
    2024-07-30
    0
    2
    • 默认21344回复GOYU\XU:

      因为我想转出 浮点型（float32）的onnx模型，我看源码：

      只有两个选择，一个是“stage == "int_infer"，是量化导出onnx模型， 我选择stage==qat，是不想进行量化导出onnx模型，只想导出浮点模型。是我考虑错了吗？需要选择stage == "int_infer吗？

      ”

      2024-07-31
      0
    • GOYU\XU回复默认21344:

      你好，目前问题有解决吗，是模型转换的问题吗？

      2024-08-06
      0
  • GOYU\XU
    Lv.2

    模型训练loss是正常下降的吗

    2024-07-29
    0
    1
    • 默认21344回复GOYU\XU:

      loss 是正常下降的，指标也是正常上升的。

      2024-07-30
      0
  • momo（社区版）
    Lv.4

    请导出float.onnx，修改config文件中的onnx_cfg，如下：

    

    2024-08-06
    0
    0