Join GitHub today
GitHub is home to over 28 million developers working together to host and review code, manage projects, and build software together.Sign up
CORe50 Dataset + reporthook for download in Utils #340
I've also added a simple report hook function (credit to Shichao) in the Utils.py file for showing up the status of the download which can be useful for big Datasets.
Note: This implementation it's not super efficient since similar to the ImageFolder class but good for small RAM devices. I plan to update the code with a more efficient pre-loading strategy in the near future.
I haven't had time to check the PR fully yet, but instead of adding a report hook function to utils why not use tqdm (if available)?
try: from tqdm import tqdm except ImportError: tqdm = lambda x:x
Then you can just wrap the loop in
You can also add something to the README/docstring about optional packages for torchvision (to enhance usability but not required to use it)
Hi @alykhantejani! Thank you for the feedback!
Maybe I misunderstood your suggestion but using
Moreover, with just a 10-lines function we can provide download details also to users who don't have