Tensorrt_Mask2Former icon indicating copy to clipboard operation
Tensorrt_Mask2Former copied to clipboard

Published 20 hours ago • verified publisher icon Luckydog-lhy

把自己训练的在ade20k数据集上的语义分割模型导出为scripting方法的模型

Open GitHubYuxiao opened this issue 1 year ago • 1 comments

如果我想我把自己训练的在ade20k数据集上的语义分割模型导出为scripting方法的模型,是不是通过MasksFormer文件夹下面的demo文件夹下的export.py脚本就可以实现

GitHubYuxiao avatar Jan 15 '24 00:01 GitHubYuxiao

如果我想我把自己训练的在ade20k数据集上的语义分割模型导出为scripting方法的模型,是不是通过MasksFormer文件夹下面的demo文件夹下的export.py脚本就可以实现

是的，如果你是使用的mask2former的标准网络结构的话

Luckydog-lhy avatar Jan 16 '24 06:01 Luckydog-lhy