从事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版本


Logo

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

更多推荐