- 博客(11)
- 资源 (3)
- 收藏
- 关注
原创 配置pytorch安装环境及一些常见的包
1.安装环境打开Anaconda Prompt(anaconda)2.查看安装的python版本python --version3.创建pytorch环境conda create -n pytorch python=3.8.44.激活环境conda activate pytorch5.镜像源的命令语句conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/c
2021-08-16 14:07:10 290
转载 torch预训练权重转换为paddle模型格式:Resnet50模型
import torchimport paddleimport osfrom collections import OrderedDictimport numpy as npdef export_weight_names(net): print(net.state_dict().keys()) with open('paddle.txt', 'w') as f: for key in net.state_dict().keys(): f.w
2021-08-14 20:13:31 993
原创 批量删除一个文件夹中与另一个文件夹名称相同的文件
删除一个文件夹中与另一个文件夹名称相同的图片,可用于机器学习或者深度学习前,图像素材的准备阶段,批量删除图片,节省时间import osdef file1(dirpath): for root, dirs, file1 in os.walk(dirpath): return file1def file2(dirpath): for root, dirs, file2 in os.walk(dirpath): return file2file1=fi
2021-08-05 14:20:31 1140
原创 修改XML文件编码格式
import osimport codecsimport chardetclass UpdateCoding: def __init__(self, base_file): file_list = os.listdir(base_file) print(file_list) for file_name in file_list: # linux 文件路径 / # window 文件路径 \
2021-07-30 17:30:11 896
原创 新增VOC数据集图片与XML文件重新命名
import numpy as npimport globimport osimport xml.etree.ElementTree as ETimport xml.dom.minidom'''第一步,将xml文件和图片重新命名'''# 获取文件夹中图片的数量def getDirImageNum(path): bmpDirImagesNum = 0 for bmpfile in os.listdir(path): if os.path.splitext
2021-07-30 16:35:56 542 2
原创 目标检测VOC数据集:用PIL实现xml和图片的同步缩放
import osfrom PIL import Imageimport platformimport xml.dom.minidomclass HandleXmlImage: def get_file_path(self, dir_file): self.system_version = platform.platform() file_name_list = os.listdir(dir_file) for file_name in
2021-05-10 09:22:26 358
原创 同步缩放XML文件与图片至256
import globimport xml.dom.minidomimport cv2from PIL import Imageimport matplotlib.pyplot as pltimport os# 定义待批量裁剪图像的路径地址IMAGE_INPUT_PATH = 'D:/Object Detection/2021V1/JPEGImages'XML_INPUT_PATH = 'D:/Object Detection/2021V1/Annotations'# 定义裁剪后的图像存
2021-05-08 18:01:43 318 1
原创 批量去除XML文件中的<?xml version=“1.0“ encoding=“utf-8“?>
#!/usr/bin/python# -*- coding: UTF-8 -*-import osdef file_name(input_dir,output_dir,spile): for root, dirs, files in os.walk(input_dir): for item in files: f = open(input_dir+item, "r",encoding='UTF-8')
2021-05-08 13:21:34 4975
原创 修改文件夹里所有xml文件的name值
import xml.dom.minidomimport ospath='D:/444/yitihuajifang2'# xml文件存放路径sv_path='D:/444/yitihuajifang'# 修改后的xml文件存放路径files=os.listdir(path)for xmlFile in files: dom=xml.dom.minidom.parse(os.path.join(path,xmlFile)) #打开xml文件,送到dom解析 root=dom.documentEl
2021-05-08 09:21:17 148
原创 根据xml找出源文件中所有jpg的文件
#根据挑选的jpg文件找出对应的xml、json、shp、dbf、shx文件import osimport shutilfilepath1 = 'D:/BaiduNetdiskDownload/tieta/Annotations/'#源文件做参考file_list = os.listdir(filepath1)print(file_list)print(len(file_list))filepath2 = 'D:/444/zhuanhunjifang'#需要拷出的文件位置filepat
2021-05-08 09:20:01 187
原创 VOC格式目标检测数据集:XML与图片互校验
# -*- coding: utf-8 -*-import ospath1 = 'D:/Object Detection/image2/'path2 = 'D:/Object Detection/Annotation/'def file_name(image_dir,xml_dir): jpg_list = [] xml_list = [] for root, dirs, files in os.walk(image_dir): for file in
2021-05-08 09:18:18 175
脚本可以批量去除XML文件中的<?xml version=“1.0“ encoding=“utf-8“?>
2021-05-10
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人