数据处理
cv等领域的数据处理方法
dear_queen
这个作者很懒,什么都没留下…
展开
-
深度学习SOTA的tricks
深度学习SOTA的tricks分享转载 2022-10-10 09:32:11 · 138 阅读 · 0 评论 -
根据csv文件划分类别文件夹
首先看下我的文件:这里train.csv为我们的训练文件目录,包含image和labels两列内容。train_images存放的是全部的训练图像数据,train存放我们即将生成的分类文件夹,目前为空。代码如下:其中label_path是训练csv文件,o_pth为文件目录,new_pth为想要存放的分类文件夹目录。import osimport timeimport shutilimport pandas as pdlabel_path = 'D:\data_test\plant-p原创 2022-04-05 22:49:33 · 2126 阅读 · 0 评论 -
推荐几个非常好用的计算flops、params、内存...的工具
推荐几个非常好用的计算flops、params、内存…的工具1、ptflops使用方法如下:import torchimport torchvision.models as models# import torchfrom ptflops import get_model_complexity_infomodel = resnet34(num_classes=5)model_weight_path ='xxxx'model.load_state_dict(torch.load(model原创 2021-04-10 17:01:05 · 1811 阅读 · 0 评论 -
VOC数据集自检 xml文件
自己制作数据集时,很有可能出现标签错误,不小心弄错等问题。1.检测数据集中的错误样本:import osimport xml.etree.ElementTree as ETimport numpy as np# np.set_printoptions(suppress=True, threshold=np.nan)import matplotlibfrom PIL import Imagedef parse_obj(xml_path, filename): tree = ET.原创 2020-11-18 15:54:08 · 436 阅读 · 3 评论 -
好用的代码合集(数据集处理)
1.coco转voc格式from pycocotools.coco import COCOimport osimport shutilfrom tqdm import tqdmimport skimage.io as ioimport matplotlib.pyplot as pltimport cv2from PIL import Image, ImageDraw# the path you want to save your results for coco to vocsavep原创 2021-09-16 15:17:51 · 860 阅读 · 0 评论