使用google colab将相同类别的图像和json文件过滤到google drive中的另一个文件夹中

from google.colab import drive
drive.mount('/content/drive')

from google.colab import files
uploaded = files.upload()

import os
import shutil
import json

# 设置输入文件夹和输出文件夹的路径
input_folder = '/content/drive/MyDrive/input_folder'
output_folder = '/content/drive/MyDrive/output_folder'

# 创建输出文件夹
os.makedirs(output_folder, exist_ok=True)

# 遍历输入文件夹中的所有文件
for file_name in os.listdir(input_folder):
    if file_name.endswith('.jpg') or file_name.endswith('.png'):
        # 图像文件
        shutil.move(os.path.join(input_folder, file_name), os.path.join(output_folder, file_name))
    elif file_name.endswith('.json'):
        # json文件
        with open(os.path.join(input_folder, file_name), 'r') as f:
            data = json.load(f)
        
        # 根据类别进行过滤，假设类别为'cat'
        if data['category'] == 'cat':
            shutil.move(os.path.join(input_folder, file_name), os.path.join(output_folder, file_name))