关于sam即segment-anythin中demo实现步骤

发布时间 2023-07-18 17:59:50作者: 海_纳百川

说实话,这个项目的readme写的真的不咋地,很乱。

我说说这个demo的部署实现步骤吧。

1.安装yarn,这个自己去安装吧

2.将demo/assets/data/dogs.jpg使用numpy读取,保存为npy文件,放在跟dogs.jpg同一目录下

image = cv2.imread('src/assets/data/dogs.jpg')
predictor.set_image(image)
image_embedding = predictor.get_image_embedding().cpu().numpy()
np.save("dogs_embedding.npy", image_embedding)

3.将pth模型转为onnx模型,放到demo/model中,注意文件名要保存对,不然读取不了

python3 scripts/export_onnx_model.py --checkpoint sam_vit_h_4b8939.pth  --model-type vit_h --output sam_onnx_quantized_example.onnx

 

4.然后使用如下开启网址

yarn
yarn start

 然后部署就是这样,说实话就是个离线网页,交互性很差