nodefluxio / vortex

A Deep Learning Model Development Framework for Computer Vision
27 stars 6 forks source link

[BUG] Dataset is not registered #83

Closed internnos closed 4 years ago

internnos commented 4 years ago

Describe the bug Dataset cannot be registered

To Reproduce Steps to reproduce the behavior:

  1. Pull latest image from docker pull nodefluxio/vortex:latest
  2. Download dataset from https://gitlab.com/nodefluxio/dataset-repository/image-classification/liveness
  3. $ vortex Expected behavior Liveness dataset is registered

Screenshots image

Additional context Add any other context about the problem here.

triwahyuu commented 4 years ago

I haven't reproduce your issue, but can I know where did you put your dataset?

internnos commented 4 years ago

You can get the dataset from here

triwahyuu commented 4 years ago

no no, I mean the path where you put the dataset inside the docker container.
currently the dataset only reads in external/datasets relative to where you call the vortex cli. you can also put your dataset directory path to an environment variable, but I forgot the name of the variable, let me look it up.

triwahyuu commented 4 years ago

the environment variable name is VORTEX_DATASET_ROOT

internnos commented 4 years ago

im sorry i got it all wrong. i already put it in external/datasets as mentioned on the docs but the vortex still skip my dataset

image

triwahyuu commented 4 years ago

ohh I see, let me try to reproduce it tonight then.

triwahyuu commented 4 years ago

I found the problem, it is your dataset that doesn't meet vortex's requirement.
To be found by vortex, your dataset definition should contain 3 things at the same file: dataset class definition, supported_dataset variable (list) which describes all of the available dataset class in the file, and create_dataset function that which returns the dataset object.

I have fixed your implementation and made a PR on it. check it out.

triwahyuu commented 4 years ago

I think we need to add support for dataset registration decorator, cc @alifahrri

internnos commented 4 years ago

@triwahyuu thanks a bunch! I'd check on it

internnos commented 4 years ago

Works like a charm. Thanks a lot!