airctic / icedata

IceData: Datasets Hub for the *IceVision* Framework
https://airctic.github.io/icedata/
Apache License 2.0
49 stars 13 forks source link

Add `dataset` function for each dataset #96

Open lgvaz opened 3 years ago

lgvaz commented 3 years ago

🚀 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)

lgvaz commented 3 years ago

Added dataset function for fridge

jpoberhauser commented 3 years ago

I can get started on coco

ganesh3 commented 3 years ago

I can do it on pets