ztransforms

Implementation of torchvision-like based on albumentation


Keywords
albumentation, data-augmentation, imgaug, torchvision, transforms
License
Apache-2.0
Install
pip install ztransforms==0.2.0

Documentation

语言: 🇨🇳 🇺🇸

«ZTransforms»是一个图像数据扩充代码库

参考pytorch/vision实现架构,以imgaug为后端实现图像数据扩充。imgaug支持的图像格式为numpy ndarray,数据类型默认为uint8,通道排列顺序为RGB

内容列表

背景

PyTorch除了拥有强大的深度学习代码库之外,还额外提供了数据集处理、模型定义以及数据转换的实现pytorch/vision。其中transforms模块默认以PIL为后端(也可以使用更快的Pillow-SIMD,如果安装的话)提供了多种数据转换功能,比如旋转、翻转、颜色抖动、随机裁剪、中心裁剪等等。不过随着深度学习的发展,越来越多的数据转换功能被发现,而PIL库并没有进行实现

在网上找到另一个数据扩充功能库imgaug,其实现了更多的数据转换函数。所以新建这个代码库,参考transforms实现方式,以imgaug为后端进行数据转换

主要维护人员

  • zhujian - Initial work - zjykzj

致谢

@misc{imgaug,
  author = {Jung, Alexander B.
            and Wada, Kentaro
            and Crall, Jon
            and Tanaka, Satoshi
            and Graving, Jake
            and Reinders, Christoph
            and Yadav, Sarthak
            and Banerjee, Joy
            and Vecsei, Gábor
            and Kraft, Adam
            and Rui, Zheng
            and Borovec, Jirka
            and Vallentin, Christian
            and Zhydenko, Semen
            and Pfeiffer, Kilian
            and Cook, Ben
            and Fernández, Ismael
            and De Rainville, François-Michel
            and Weng, Chi-Hung
            and Ayala-Acevedo, Abner
            and Meudec, Raphael
            and Laporte, Matias
            and others},
  title = {{imgaug}},
  howpublished = {\url{https://github.com/aleju/imgaug}},
  year = {2020},
  note = {Online; accessed 01-Feb-2020}
}

参与贡献方式

欢迎任何人的参与!打开issue或提交合并请求。

注意:

许可证

Apache License 2.0 © 2021 zjykzj