#55 · Primary category: Computer Vision

imgaug

affine-transformation augment-images augmentation bounding-boxes contrast crop deep-learning heatmap image-augmentation images keypoints machine-learning polygon segmentation-maps

Image augmentation for machine learning experiments.

Project last updated:07/30/24

GitHub Stars

14.7K

Forks

2.5K

Contributors

36

License

MIT

Why we included this project

Training robust vision models usually means giving the model more variety than your raw dataset provides, and imgaug is one of the more complete toolkits for generating that variety. It applies a wide range of augmentations, from geometric transforms like rotation and cropping to noise, contrast, and color shifts, and it keeps annotations in sync as it goes. Bounding boxes, keypoints, polygons, heatmaps, and segmentation maps stay aligned with the altered images, which matters for object detection, pose estimation, and segmentation pipelines where a naive crop or flip would silently corrupt the labels. Teams running custom PyTorch or TensorFlow training loops can use it as a preprocessing stage without reworking their data flow. It is well documented and a dependable choice for reproducible, label-aware data expansion in computer vision experiments.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category