MCPcopy Index your code
hub / github.com/cuitno1/Data-enhancement

github.com/cuitno1/Data-enhancement @main

Chat with this repo
repository ↗ · DeepWiki ↗ · + Follow
30 symbols 80 edges 4 files 11 documented · 37%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

Data-enhancement

2022/10/22 修改了github代码导致xmax<xmin的情况。(由于上传疏忽,正确的代码修改了后上传的错误的版本)

Support data enhancement when there are few data sets(支持数据集较少的情况进行数据增强,包含随机的多种变化) 这是数据集扩增的一个小工具,在您想使用yolo等目标检测算法时数据集较少的情况下能够进行变化增强图片,丰富您的数据集。 (支持LabelIMg和LabelMe标注的文件) 包括3个python文件

(rename_file.py、DataAugmentforLabelImg.py和DataAugmentforLabelMe.py) 原文链接:https://blog.csdn.net/qq_39740357/article/details/123258697

  1. rename_file.py能实现文件的重命名,注意修改文件的路径
  2. DataAugmentforLabelImg.py能实现LabelImg标注后的图片的增强(包括模糊,亮度,裁剪,旋转,平移,镜像等变化)
  3. DataAugmentforLabelMe.py能实现LabelMe标注后的图片的增强(包括模糊、亮度、平移、镜像等变化)

注意:一些包的安装是必要的,比如Opencv_python等

将您需要增强的图片放在对应的文件夹即可,具体可参考demo给出的图片和xml文件存放路径,您放入即可

保存结果:

  1. 目标检测增强后的图片默认会保存在./data/Images2中,xml文件会保存在./data/Annotations2中(包括新的图片和xml文件)

  2. 目标分割增强后的图片默认会保存在./data2中(包括新的图片和json文件)

实现这个工具参考了:

https://github.com/maozezhong/CV_ToolBox/blob/master/DataAugForObjectDetection/

Core symbols most depended-on inside this repo

deal
called by 1
Data-enhancement/rename_file.py
_addNoise
called by 1
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py
_changeLight
called by 1
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py
_addRandPoint
called by 1
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py
_shift_pic_bboxes
called by 1
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py
_filp_pic_bboxes
called by 1
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py
dataAugment
called by 1
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py
parse_json
called by 1
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py

Shape

Method 24
Class 4
Function 2

Languages

Python100%

Modules by API surface

Data-enhancement/DataAugForObjectDetection/DataAugmentforLabelImg.py16 symbols
Data-enhancement/DataAugForObjectSegmentation/DataAugmentforLabelMe.py13 symbols
Data-enhancement/rename_file.py1 symbols

For agents

$ claude mcp add Data-enhancement \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact