icedata icon indicating copy to clipboard operation
icedata copied to clipboard

Add `dataset` function for each dataset

Open lgvaz opened this issue 3 years ago • 3 comments

🚀 Feature

For each dataset we can also add a dataset function that returns a train,valid dataset with default transforms included.

This would also reduce the amount of repeated code on icevision tutorials.

HOW TO

Follow the structure of datasets/fridge/dataset.py or datasets/pennfudan/dataset.py:

  1. Create dataset.py file
  2. Add from icedata.datasets.<DATASET_FOLDER>.dataset import * to datasets/<DATASET_FOLDER/__init__.py

TODO (no specific order)

  • [x] fridge
  • [x] pennfudan
  • [x] birds
  • [x] biwi
  • [x] coco (In progress by: @jpoberhauser)
  • [ ] ochuman
  • [x] pets (in progress by: @ganesh3)
  • [x] voc

lgvaz avatar Mar 10 '21 16:03 lgvaz

Added dataset function for fridge

lgvaz avatar Mar 12 '21 13:03 lgvaz

I can get started on coco

jpoberhauser avatar Mar 17 '21 18:03 jpoberhauser

I can do it on pets

ganesh3 avatar Mar 18 '21 02:03 ganesh3