augment only a single, or set of images about augmentor HOT 3 OPEN

@mdbloice, Thanks for this awesome library. Just wondering similar to transforming the mask, is it possible to transform the bounding box coordinates as well ?

from augmentor.

Hi @niekverw , yes, this is now finally possible using the ground_truth() function in the Pipeline class (as of Augmentor v0.2.0).

Basically, just use this function to point to a directory containing the mask images, and when you generate new image data, images will be generated for both the mask and the original images:

p = Augmentor.Pipeline("/home/user/images")
p.ground_truth("/home/user/mask_images")

Call p.sample() in the same way as per usual to generate both sets of images.

See the README for more details.

from augmentor.

@mdbloice Hi there, Can you explain more about masks in the ground truth directory? I have image files and annotation files (a text file containing coordinates and dimensions of bounding boxes). My interest is to apply augmentation on image (shift, rotation, or flip), but also similarly to grounding boxes (masks?). Thanks.

from augmentor.