search

oke-aditya / quickvision

An Easy To Use PyTorch Computer Vision Library
Apache License 2.0
51 stars 4 forks source link

[RFC] Data Augmentation of object detection #83

Open zhiqwang opened 3 years ago

zhiqwang commented 3 years ago

🚀 Feature

Hi, Recently I found a repo about data augmentation with excellent programming, named TensorPipe. But it's implemented with tensorflow, do you mind to add this similar features based in PyTorch in our quickvision repo?

Motivation

Augmentation play an important role in practical application especially when the collection of images and its labels is restricted by various reasons or the cost is very high.

Pitch

oke-aditya commented 3 years ago

Yes we can have such augmentation. Especially bounding box augmentations. As of now torchvision itself is getting upgrade for transforms. So maybe after a few releases we will have this.

We can plan for native ops module too.

oke-aditya commented 3 years ago

Once we have a nice nested tensor support in PyTorch as pointed out in torchvision (vmap, etc) It wil be possible for torchvision itself to have params such as probablistic transforms, keep some batches from transforms. This will make transforms even more powerful.

zhiqwang commented 3 years ago

That will be cool!