Fixed issues with dataset loading and input_boxes #495
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Dataset loading was giving this error:
NotImplementedError: Loading a dataset cached in a LocalFileSystem is not supported
This was solved by simply adding :
pip install -U datasets
Also another error was encountered when passing input_boxes to the preprocesser:
inputs = processor(image, input_boxes=[input_boxes], return_tensors="pt").to(device)
ValueError: Input boxes must be a list of list of list of floating points.
Which was solved by adding an extra square brackets
inputs = processor(image, input_boxes=[[input_boxes]], return_tensors="pt").to(device)