Open lgvaz opened 3 years ago
For each dataset we can also add a dataset function that returns a train,valid dataset with default transforms included.
dataset
train,valid
This would also reduce the amount of repeated code on icevision tutorials.
Follow the structure of datasets/fridge/dataset.py or datasets/pennfudan/dataset.py:
datasets/fridge/dataset.py
datasets/pennfudan/dataset.py
dataset.py
from icedata.datasets.<DATASET_FOLDER>.dataset import *
datasets/<DATASET_FOLDER/__init__.py
Added dataset function for fridge
I can get started on coco
I can do it on pets
🚀 Feature
For each dataset we can also add a
dataset
function that returns atrain,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
ordatasets/pennfudan/dataset.py
:dataset.py
filefrom icedata.datasets.<DATASET_FOLDER>.dataset import *
todatasets/<DATASET_FOLDER/__init__.py
TODO (no specific order)