添加图片转为支持yolov7格式的工具

需要使用opencv
This commit is contained in:
xiaoyuluoke 2022-10-12 13:08:24 +00:00
parent 27d82c6652
commit f51c5a7e8b
1 changed files with 20 additions and 0 deletions

20
img2yolov7.py Normal file
View File

@ -0,0 +1,20 @@
import os
import cv2
bmp_dir = r'.'
jpg_dir = r'png'
def file_name(file_dir):
for root, dirs, files in os.walk(file_dir):
count = 1
# 当前文件夹所有文件
for i in files:
# 判断是否以.jpg结尾
if i.endswith('.bmp'):
# 如果是就改变图片像素为640,480
img = cv2.imread(os.path.join(bmp_dir,i),-1)
newName = i.replace('.bmp','.png')
im2 = cv2.resize(img,(640,640),) # 为图片重新指定尺寸
cv2.imwrite(os.path.join(jpg_dir,newName),im2)
print('转变图:%s'%newName)
break
file_name('./') # 当前文件夹