data/owlvit-base-patch32-image-encoder.engine" not found

[enazari]

Hi @jaybdub ,
I have a question about populating the data folder with different engine files. How should one populate this folder? After installing all the dependencies, when I run the script below, I encounter FileNotFoundError error:

from nanoowl.owl_predictor import OwlPredictor

predictor = OwlPredictor(
    "google/owlvit-base-patch32",
    image_encoder_engine="data/owlvit-base-patch32-image-encoder.engine"
)

image = PIL.Image.open("assets/owl_glove_small.jpg")

output = predictor.predict(image=image, text=["an owl", "a glove"], threshold=0.1)

print(output)

The error is:
FileNotFoundError: [Errno 2] No such file or directory: 'data/owlvit-base-patch32-image-encoder.engine'

Thank you very much for your patience and help.

Best,
Ehsan

[jaybdub]

Hi @enazari ,

The path should be relative to the location your running the Python script from.

Which directory are you executing the code from, and where is the engine file relative to this?

John

[enazari]

Thank you @jaybdub for your prompt reply.

I created a jupyter file inside nanoowl directory where data directory lies. Inside data directory, I have owl_image_encoder_patch32.engine and owl_predict_out.jpg that are generated when I executed other scripts in this repo's readme.

I am not sure how I can generate/obtain owlvit-base-patch32-image-encoder.engine.

Thank you,
Ehsan

[kenJPG]

Hi, after toying with this project, my perspective is that owlvit-base-patch32-image-encoder.engine is intended to be owl_image_encoder_patch32.engine. Thus you just need to rename and it will be solved.