AI常用库(数据增强,数据集处理)
·
从事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版本
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)