labelme去除空圖片

阳光天气發表於2024-05-17

Labelme是一個用於影像標註的開源工具。在使用Labelme進行資料標註後，可能會生成一些空的影像檔案（即沒有進行標註的圖片），這些空圖片通常不應該被使用。以下是一個簡單的Python指令碼，用於檢測和刪除這些空的標註檔案：

點選檢視程式碼 
import os
import json


def is_image_empty(image_path):
    # 載入JSON檔案
    json_file = image_path.replace('.jpg', '.json')
    if not os.path.isfile(json_file):
        return True
    with open(json_file, 'r') as f:
        data = json.load(f)
    # 檢查是否有標註
    return len(data['shapes']) == 0


def remove_empty_images(image_dir):
    for filename in os.listdir(image_dir):
        if filename.endswith('.jpg'):
            image_path = os.path.join(image_dir, filename)
            if is_image_empty(image_path):
                # 刪除圖片和對應的JSON檔案
                os.remove(image_path)
                json_file = image_path.replace('.jpg', '.json')
                if os.path.isfile(json_file):
                    os.remove(json_file)


# 使用方式：將image_dir替換為你的影像資料夾路徑
remove_empty_images(r'D:\pic\zhongwei\label_pic_wfdkl_wfxkl_reworded\target')

相關文章