Skip to content

[Bug] The new image transforms need to download the model libraries at run-time #1530

@shahrokhDaijavad

Description

@shahrokhDaijavad

Search before asking

  • I searched the issues and found no similar issues.

Component

transdforms/Other

What happened + What you expected to happen

At the moment, the 3 model files needed by the image transforms are downloaded a priory into a models subfolder. This won't work when using these transforms by pip installing them in a notebook. Instead, a new model_type (yolo) in model_loader should be added and these models should be downloaded at run-time.

Reproduction script

  1. pip install any of the 3 image transforms in a notebook
  2. Using them results in a model not found error.

Anything else

See above.

OS

MacOS

Python

3.12

Are you willing to submit a PR?

  • Yes I am willing to submit a PR!

Metadata

Metadata

Assignees

Labels

enhancementNew feature or request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions