图像分割模型npu适配部署分享

[lxx1236]

我们分别在310P上部署了Segment anything 和Cutie两个物体分割模型，成功在310P上部署。下面分别是我们对官方代码/环境做的一些修改，供参考：

• Segment anything
  • 官方代码torchvision找不到nms库，根据该API文档 https://ascend.github.io/docs/sources/pytorch/api_doc.html#torch_npu.npu_nms_v4，采用torch_npu.npu_nms_v4()替代。

• Cutie
  • 引入torch_npu和transfer_to_npu，正常部署即可，无需额外修改。
• 

 

[2301_80744354]

感谢分享！👍🏻