データ拡張スクリプト：pythonは、任意のサイズのネガティブサンプル画像を使用して、トレーニングに使用できる空のxmlファイルのバッチを作成します

使用する：

空のxmlネガティブサンプルは、ターゲット検出テストプロセス中の誤検出率を減らすために使用されます。

使用法：

ネガティブサンプル画像とトレーニングセット画像は同じフォルダに配置され、空のxmlファイルとトレーニングセットxmlは同じフォルダに配置され、通常のターゲット検出トレーニングプロセスに従ってトレーニングを完了することができます。

追加機能：

元のラベルをバックグラウンドxmlファイルとして置き換えます（主に、競合データセットのバックグラウンドラベルを検索および変更するために使用されます）。

コード上（入力フォルダーと出力フォルダーを独自のものに変更します）：

import os
import xml.dom.minidom
import cv2
 
img_path = 'C:\\Users\\hq\\Desktop\\none_test\\img\\'
xml_path = 'C:\\Users\\hq\\Desktop\\none_test\\xml\\'

# 删除所有label为背景的xml
import xml.etree.ElementTree as ET
for img_file in os.listdir(xml_path):
    filename = os.path.join(xml_path, img_file)
    root = ET.parse(filename).getroot()
    for ob in root.findall('object'):
        name = ob.find('name').text
        #print(name)
    if name=='Background':
        os.remove(filename)
        continue

# 创建新的xml文件（保留所有原有label为非背景的xml）
for img_file in os.listdir(img_path):    
    filename = os.path.join(img_path, img_file)
    img_cv = cv2.imread(filename)
    
    img_name = os.path.splitext(img_file)[0]
 
    #create an empty dom document object
    doc = xml.dom.minidom.Document()
    #creat a root node which name is annotation
    annotation = doc.createElement('annotation')
    #add the root node to the dom document object
    doc.appendChild(annotation)
 
    #add the folder subnode
    folder = doc.createElement('folder')
    folder_text = doc.createTextNode('VOC2012')
    folder.appendChild(folder_text)
    annotation.appendChild(folder)
 
    #add the filename subnode
    filename = doc.createElement('filename')
    filename_text = doc.createTextNode(img_file)
    filename.appendChild(filename_text)
    annotation.appendChild(filename)
 
    # add the path subnode
    path = doc.createElement('path')
    path_text = doc.createTextNode(img_path + img_file)
    path.appendChild(path_text)
    annotation.appendChild(path)
 
    #add the source subnode
    source = doc.createElement('source')
    database = doc.createElement('database')
    database_text = doc.createTextNode('Unknown')
    source.appendChild(database)
    database.appendChild(database_text)
    annotation.appendChild(source)
 
    #add the size subnode
    size = doc.createElement('size')
    width = doc.createElement('width')
    width_text = doc.createTextNode('%s'%img_cv.shape[1])
    height = doc.createElement('height')
    height_text = doc.createTextNode('%s'%img_cv.shape[0])
    depth = doc.createElement('depth')
    depth_text = doc.createTextNode('%s'%img_cv.shape[2])
    size.appendChild(width)
    width.appendChild(width_text)
    size.appendChild(height)
    height.appendChild(height_text)
    size.appendChild(depth)
    depth.appendChild(depth_text)
    annotation.appendChild(size)
 
    #add the segmented subnode
    segmented = doc.createElement('segmented')
    segmented_text = doc.createTextNode('0')
    segmented.appendChild(segmented_text)
    annotation.appendChild(segmented)
 
    #write into the xml text file
    #os.mknod(xml_path+'%s.xml'%img_name) 
    if not os.path.exists(xml_path+'%s.xml'%img_name):
        with open(xml_path+'%s.xml'%img_name, mode="w", encoding="utf-8") as f:
            fp = open(xml_path+'%s.xml'%img_name, 'w+')
            doc.writexml(fp, indent='\t', addindent='\t', newl='\n', encoding='utf-8')
            fp.close()

結果を生成します。

最後に：この大物ブログhttps://blog.csdn.net/dulingwen/article/details/89669928のインスピレーションをありがとうございます