/tools_pytorch

pytorch tools like image argumentor and load etc.

Primary LanguagePython

tools_pytorch

这是保存在做计算机视觉/深度学习/机器学习过程中比较有用的代码,深度学习框架用的pytorch

index

  • Image_Annotation_Augment.py 目标检测数据增强
  • make_imageset.py 分配训练集/测试集/验证集 为VOC标准格式
  • xml_helper.py xml文件的读取与生成
  • data_augmentor_generate.py 图像数据增强(不包括标注,适用于图像分类)
  • calcu_mean.py 计算图像各通道均值,用于预处理
  • Map_calcu 目标检测结果的指标计算