vislab
vislab copied to clipboard
How to load the features from a dataset end-to-end?
Hi Sergey - I noticed in /vislab/feature.py there is a compute command which appears to load the features. I'd just like to confirm the end-to-end procedure of adding my own dataset and getting features into the db.
So - follow the "Adding your own" section of the dataset documentation:
- Add the file vislab/datasets/your_dataset.py that will contain a function to load a pandas.DataFrame with:
- unique string-based index, with name image_id
- image_url or image_filename in columns
- a column for whatever boolean label you care about (does this matter if I am just extracting neural codes?)
- Modify DATASETS in vislab/dataset.py to map a name to your new function.
- Then finally, call feature.py compute from command line? (is feature.py the main entry point?)