diff --git a/demo/README.md b/demo/README.md index 41f1ddd80..cce3018db 100644 --- a/demo/README.md +++ b/demo/README.md @@ -40,7 +40,7 @@ Set the new image and export the embedding: ``` image = cv2.imread('src/assets/dogs.jpg') -predictor.set_image(image) +predictor.set_image(image, "BGR") image_embedding = predictor.get_image_embedding().cpu().numpy() np.save("dogs_embedding.npy", image_embedding) ```