используйте openAI CLIP с тензором факела или массивом numpy в качестве входных данных

Question

Как правильно использовать тензор pytorch или iamge openCV в качестве входных данных для OpenAi CLIP?

Я пробовал следующее, но пока это не сработало:

      device = "cuda" if torch.cuda.is_available() else "cpu"
clip_model, clip_preprocess = clip.load("ViT-B/32", device=device)
clip_preprocess(torch.from_numpy(OpenCVImage)).unsqueeze(0).to(device)

• шаг предварительной обработки завершается с ошибкой с сообщениемProcess finished with exit code 137 (interrupted by signal 9: SIGKILL)
• openCVImage — это объект, который уже был обработан с помощью cv2.COLOR_BGR2RGB.

Answer 1

Похоже на системную проблему(SIGKILLпрерывать).
Какую систему вы используете? Какая ОС? Какой типdevice? Возможно ли, что вы превысили лимит памяти CPU/GPU?