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

说实话，这个项目的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

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