EdVince / CLIP-ImageSearch-NCNN

CLIP⚡NCNN⚡基于自然语言的图片搜索(Image Search)⚡以字搜图⚡x86⚡Android
224 stars 21 forks source link

模型是怎么导出的? #8

Open perfyperfect opened 1 year ago

perfyperfect commented 1 year ago

我尝试导出clip的模型(vit和Resnet都尝试了)分别都报错了,请问您能分享一下导出模型的经验么?

EdVince commented 1 year ago
  1. 导出resnet要比导出vit的要简单,你可以先从导出resnet开始。
  2. 如果导出经验不多的话,就用pnnx转模型,不要用onnx了。
  3. clip有text encoder和image encoder,这两个要分开导出,不要一起导。