Kc

转模型报错

2

执行代码: def export_embedding_model(): sam_checkpoint = "D:/Anaconda3/envs/sam_vit_h_4b8939.pth" model_type = "vit_h" device = "cpu" sam = sam_model_registry[model_type](checkpoint=sam_checkpoint) sam.to(device=device) image = cv2.imread('./images/truck.jpg') target_length = sam.image_encoder.img_size pixel_mean = sam.pixel_mean pixel_std = sam.pixel_std img_size =...