icedata
icedata copied to clipboard
Add `dataset` function for each dataset
🚀 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
:
- Create
dataset.py
file - Add
from icedata.datasets.<DATASET_FOLDER>.dataset import *
todatasets/<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
Added dataset
function for fridge
I can get started on coco
I can do it on pets