Skip to content

Support for imgaug #1954

@5starkarma

Description

@5starkarma

🚀 Feature

imgaug is very popular and is easy to work with. I am wondering if I put in PR's which allowed use of imgaug, such as BoundingBox instances, to be used in parts of the program if it would be approved.

e.g. The plot_one_box function could accept then convert BoundingBox instances to the proper format (tensor) for easy plotting of bboxes.

Example:

def plot_one_box(x, img, color=None, label=None, line_thickness=None):
    # If bbox is instance of BoundingBox
    if isinstance(x, BoundingBox):
        # Convert to tensor
        x = torch.tensor([x.x1, x.y1, x.x2, x.y2])

Motivation

I use imgaug, albumentations uses imgaug. It is very easy to use.

Pitch

Add functionality which allows use of imgaug.

Metadata

Metadata

Assignees

No one assigned

    Labels

    StaleStale and schedule for closing soonenhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions