使用我自己的图像在拥抱脸部视觉伯特上使用的演示笔记本的麻烦

Question

我使用自己的图像与此。我想上传自己的图像，而不是使用预定义的URL链接。例如，在此笔记本电脑中，我们使用以下代码：

URL = "https://vqa.cloudcv.org/media/test2014/COCO_test2014_000000262567.jpg"
frcnn_cfg = Config.from_pretrained("unc-nlp/frcnn-vg-finetuned")
frcnn = GeneralizedRCNN.from_pretrained("unc-nlp/frcnn-vg-finetuned", config=frcnn_cfg)
image_preprocess = Preprocess(frcnn_cfg)
images, sizes, scales_yx = image_preprocess(URL)
output_dict = frcnn(
    images,
    sizes,
    scales_yx=scales_yx,
    padding="max_detections",
    max_detections=frcnn_cfg.max_detections,
    return_tensors="pt",
)

因此，它为我们提供了功能，但是我想在以下行中实现某些内容：

# img (already being uploaded to google colab via local machine)

frcnn_cfg = Config.from_pretrained("unc-nlp/frcnn-vg-finetuned")
frcnn = GeneralizedRCNN.from_pretrained("unc-nlp/frcnn-vg-finetuned", config=frcnn_cfg)
image_preprocess = Preprocess(frcnn_cfg)

images, sizes, scales_yx = image_preprocess(img)

output_dict = frcnn(
    images,
    sizes,
    scales_yx=scales_yx,
    padding="max_detections",
    max_detections=frcnn_cfg.max_detections,
    return_tensors="pt",
)

有什么办法可以实现它？