lightly-ai / lightly

A python library for self-supervised learning on images.
https://docs.lightly.ai/self-supervised-learning/
MIT License
3.16k stars 284 forks source link

Read dataset object-wise #412

Closed MalteEbner closed 3 years ago

MalteEbner commented 3 years ago

Description

For object detection tasks, it should be possible to read a dataset object-wise, do the embedding on the objects cropped from the full image and then upload the object to the webapp.

Proposed workflow:

Open questions

IgorSusmelj commented 3 years ago

Something like this?

We expect the dataset to be in the Yolo format:

`- data_root
  `- images
    `- img_001.jpg
    `- img_002.jpg
  `- labels
    `- img_001.txt
    `- img_002.txt

Then a user can run lightly-crop:

lightly-crop images_dir=data_root/images labels_dir=data_root/labels output_dir=data_root/images_cropped padding=0.1

That would translate into

MalteEbner commented 3 years ago

Ideally use the dataset for MilkyWay for it: https://github.com/eg4000/SKU110K_CVPR19