omoindrot / tensorflow-triplet-loss

Implementation of triplet loss in TensorFlow
https://omoindrot.github.io/triplet-loss
MIT License
1.12k stars 283 forks source link

Dataset building #16

Closed FSet89 closed 6 years ago

FSet89 commented 6 years ago

I would like to train the model on my own dataset. Where can I find instructions on how to build the dataset?

omoindrot commented 6 years ago

See #8

FSet89 commented 6 years ago

I am still a bit confused on the folder structure for this specific project. Is it sufficient to build a TF dataset with a list of images and a label for each image? What is the format for the labels?