likyoo/open-cd

模型导出

nzbanana opened this issue · 1 comments

nzbanana commented

请问训练完成后的模型该如何导出成通用模型呢，例如onnx格式或者torchscript格式。
我尝试了使用了TPP项目（https://github.com/KyanChen/TTP）进行训练和导出，但TTP的项目中导出模型的脚本似乎无法正常导出，open-cd是否支持模型导出？

GreatV commented

可以试试 tools/export.py