AI常用库（数据增强，数据集处理）

最新推荐文章于 2024-08-23 11:03:54 发布

One橙序猿

最新推荐文章于 2024-08-23 11:03:54 发布

阅读量473

点赞数 1

分类专栏： AI python库

本文链接：https://blog.csdn.net/rush9838465/article/details/118406354

版权

AI python库专栏收录该内容

1 篇文章 0 订阅

订阅专栏

博主分享了一个个人维护的AI辅助工具库acv，包含数据集处理、数据增强、XML解析和Excel操作等功能。已更新数据增强模块，如翻转、旋转等，并提供了VOC数据集加载及优化器如余弦退火SGD。项目已上传至GitHub，欢迎参与和贡献。

摘要由CSDN通过智能技术生成

从事CV工作也有些时间了，发现有时候会写很多小脚本处理一些琐碎的事，经常会有重复的功能，比如处理数据集，数据增强，excel处理等等。今天心血来潮准备自己定期维护一个这样的小脚本库，会不定期的更新再github上，欢迎大家来参与和纠正。

github地址：

https://github.com/rush9838465/acv

库安装方法：

pip install git+https://github.com/rush9838465/acv

更新日志：

# acv
AI 常用的功能函数，包括数据集处理，数据增强等等

# install
pip install git+https://github.com/rush9838465/acv

# logs
2021/7/2: 

    添加数据增强：flip, rotation, hsv, crop, blur, brightness, contrast, resize and padding
    调用位置： acv.data_augmentation.transforms.py

    添加数据xml解析：解析xml到json格式；从xml中裁剪bbox; VOC数据集转yolov5格式
    调用位置： acv.datasets.xml_parsing.py

    添加excel解析：读写excel文件
    调用位置： acv.excel.base.py

2021/7/5:
    
    添加数据集加载（支持torch DataLoader）：VOC数据集格式

2021/7/26:
    
    添加优化器：余弦退火 SGD版本