python - 封装moondream（备份）
目录
一、 moondream封装成类
二、moondream封装成http api
一、 moondream封装成类
# moondream_model.py
#https://github.com/vikhyat/moondream?tab=readme-ov-file
import moondream as md
from PIL import Image
class MoondreamModel:
def __init__(self, model_path):
# 初始化 Moondream 模型，加载本地模型文件
self.model = md.vl(model=model_path)
def encode_image(self, image_path):
# 加载并处理图片
image = Image.open(image_path)
# 对图片进行编码
encoded_image = self.model.encode_image(image)
return encoded_image
def generate_caption(self, encoded_image):
# 生成