提取圖片代碼:
# -*- coding: UTF-8 -*-
# !/usr/bin/env python
import sys
import re
from PIL import Image
sys.path.append('F:\\beifen')
import numpy as np
data = []
for line in open("F:\\beifen\keras-yolo3-master\VOC2018\ImageSets\Main/testjpg.txt", "r"): # 設(shè)置文件對象并讀取每一行文件
data.append(line)
for a in data:
im = Image.open('F:\\beifen\keras-yolo3-master\VOC2018\JPEGImages/{}'.format(a[:-1])) # 打開改路徑下的line3記錄的的文件名
im.save('F:\\beifen\keras-yolo3-master\VOC2018\ImageSets\Layout/{}'.format(a[:-1])) # 把文件夾中指定的文件名稱的圖片另存到該路徑下
im.close()
提取XML文件代碼:
import os
import shutil
if __name__ == '__main__':
f = open("testxml.txt","r") #存放有XML文件名字的txt
line = f.readline()
line = line[:-1]
while line:
line = f.readline()
line = line.strip('\n')
print(line)
path = os.getcwd()
new_path = "F:\\beifen\keras-yolo3-master\VOC2018\Annotations\\"+line #路徑為保存XML文件的文件夾
print(new_path)
try:
shutil.move(new_path, 'F:\\beifen\\keras-yolo3-master\\VOC2018\\123') #提取后保存的位置
except:
print("Not find error.")
# print(path)
f.close()