cwerner / fastclass

Little tools to download and then weed through images, delete and classify them into groups for building deep learning image datasets (based on crawler and tkinter)
Apache License 2.0
133 stars 25 forks source link

Provide labels in fcc #31

Open 1ycx opened 4 years ago

1ycx commented 4 years ago

I think it would be better if we can provide custom labels as input in the fcc, which if not provided defaults to the usual 1, 2, 3, 4, etc

cwerner commented 4 years ago

Sure, my initial thoughts were to keep this simple and have labels be equivalent to the number key input. Since the GUI now also features buttons for mouse input it’s more feasible to add this...

I use fcc actually for multiple purposes:

The idea is to have a way to create different subsets of the data by filtering on this later if needed (I,e, include only the best images, etc.).

One could also use it for plain classification (as I think you have in mind), however for me this is already achieved by giving multiple lines (each line for one class) to fcd

I’m not sure I can work on this the next week(s), however I’m happy to accept a PR that lets the user edit the button content to represent classes 😉

1ycx commented 4 years ago

I wanted to see If I can contribute in any way. Like a quick change of few lines but then I see there's classes and class methods which I have little experience with ( I use functions normally ) and I was like "continue this maybe sometime later". Also I have little experience with tkinter 😅

1ycx commented 4 years ago

I was doing the fastai-v3 course on deep learning and there arose the need to to create a dataset

cwerner commented 4 years ago

No worries. I like the idea... I might give it the shot this weekend actually. Not sure how hard it is to implement this with tk 🤔 I’m no expert either and it still feels pretty archaic to me...

However, as said you might be able to achieve this with fcd alone (if your search queries are for each individual class) and only use fcc to remove bad/ wrong images after the download...

cwerner commented 4 years ago

One could simply read a small config file for naming the buttons, but I’d aim to have this as an interactive operation within the GUI itself...