search

PaddlePaddle / PaddleSeg

Easy-to-use image segmentation library with awesome pre-trained model zoo, supporting wide-range of practical tasks in Semantic Segmentation, Interactive Segmentation, Panoptic Segmentation, Image Matting, 3D Segmentation, etc.
https://arxiv.org/abs/2101.06175
Apache License 2.0
8.62k stars 1.68k forks source link

适合于paddleseg的SwinTransformer预训练模型？ #1649

Closed geoexploring closed 2 years ago

geoexploring commented 2 years ago

请问，有没有适合于paddleseg的SwinTransformer预训练模型？如类似 https://bj.bcebos.com/paddleseg/dygraph/hrnet_w18_ssld.tar.gz .

谢谢！

juncaipeng commented 2 years ago

目前还没有支持swintransformer，不过我们支持了如下三个transform分割模型：

image

请参考：https://github.com/PaddlePaddle/PaddleSeg/blob/release/2.3/docs/model_zoo_overview.md#transformer%E7%B3%BB%E5%88%97

geoexploring commented 2 years ago

@juncaipeng .谢谢！ 请问，如果用paddleclas中的swintransformer ImageNet 预训练模型库可以吗？

juncaipeng commented 2 years ago

可以的，paddleclass中的swintransformer可以作为backbone。但是还需要在paddleseg中修改代码，添加模型。