yangxianpku / model_cryptor

深度学习模型加解密工具
35 stars 7 forks source link

yolov8的pt,能否加密,并且调用使用 #2

Open monkeycc opened 10 months ago

monkeycc commented 10 months ago
import cv2
from ultralytics import YOLO

# Load a pretrained YOLOv8n model
model = YOLO('yolov8n.pt')

# Read an image using OpenCV
source = cv2.imread('path/to/image.jpg')

# Run inference on the source
results = model(source)  # list of Results objects
yangxianpku commented 10 months ago

可以的吧

monkeycc commented 10 months ago

测试过不行 可能要调整

yolov8是通过文件方式读取模型的 直接给解密后的模型 就会报错

yangxianpku commented 10 months ago

更改一下读取接口应该就好了