search

hustvl / TopFormer

TopFormer: Token Pyramid Transformer for Mobile Semantic Segmentation, CVPR2022
Other
373 stars 42 forks source link

TNN tnnproto #25

Open vozhuo opened 1 year ago

vozhuo commented 1 year ago

我用其他的训练模型转换后都能在手机上跑通，唯独SegFormer和Semantic FPN ConvMLP-S有问题。请问这两个模型的tnnproto文件或者在转换过程中有经过特殊处理吗，是否可以提供tnnproto文件。

speedinghzl commented 1 year ago

@vozhuo 我们有对一些TNN不支持的操作做了些处理，简单替换成相同功能和相同计算量的操作，比如改变softmax中dim等。 tnn_models.zip 压缩包里是SegFormer和Semantic FPN ConvMLP-S的tnnproto文件，希望对你有帮助。

jhtwosun commented 1 year ago

Hi, Thanks for great work! Can you share other tnn_proto models?