Skip to content

russianhackteam/i-love-data

Repository files navigation

i-love-data

General approach on collecting custom image datasets for classification purpouses

  1. Search for prepared datasets:

  1. Search for prepared datasets:

  1. Awesome links:

ML links:

  1. Tool for object detections:

  • Classical problem is when you can google for some pictures of your interest (like "can of cola") and the picture you find contains several objects (different cans). So you need to somehow preprocess source image and extract regions of your interest. This is where pretrained YOLO or Retina may be useful (or your custom last trained dense layer using transfer learning). Implementation can be found here

  • Instead of using the raw library we had written a simple wrapper that accepts directories with pictures and target classes you want to extract. Assuming you want to extract cans from some folder with pictures:

cans

After we apply wrapper to the source image we can get seaparate cans and after it feed it directly to the network. box

  1. Annotation:

  • labelme is a super simple python module with gui for any annotation task: classification, segmentation, bb-boxes and more.

About

Dataset collection and classificator retraining

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages