2025-9-23

test命令

CUDA_VISIBLE_DEVICES=1 python train_net.py \
  --num-gpus 1 \
  --config-file configs/foodseg103/semantic-segmentation/maskformer2_R50_bs16_160k.yaml \
  --eval-only \
  MODEL.WEIGHTS output/model_final.pth \
  DATASETS.TEST "('foodseg_test',)"

对文件夹进行修改：

import os
import shutil
import hashlib
from PIL import Image
import sys

def get_file_hash(file_path):
    """计算文件的MD5哈希值，用于比较文件是否相同"""
    hash_md5 = hashlib.md5()
    with open(file_path, "rb") as f:
        for chunk in iter(lambda: f.read(4096), b""):
            hash_md5.update(chunk)
    return hash_md5.hexdigest()

def convert_to_jpeg(image_path, output_path):
    """将图片转换为JPEG格式"""
    try:
        img = Image.open(image_path)
        if img.mode != 'RGB':
            img = img.convert('RGB')
        img.save(output_path, 'JPEG', quality=95)
        return True
    except Exception as e:
        print(f"转换图片失败 {image_path}: {e}")
        return False

def main():
    # 检查是否提供了正确的参数
    if len(sys.argv) != 2:
        print("请提供主文件夹路径")
        print("使用方法: python image_organizer.py <主文件夹路径>")
        sys.exit(1)
    
    # 主文件夹路径（从命令行参数获取）
    main_folder = sys.argv[1]
    
    # 子文件夹路径
    sub_folder = os.path.join(main_folder, "可用图片-网上收集")
    
    # 检查文件夹是否存在
    if not os.path.exists(main_folder):
        print(f"主文件夹不存在: {main_folder}")
        sys.exit(1)
    
    if not os.path.exists(sub_folder):
        print(f"子文件夹不存在: {sub_folder}")
        sys.exit(1)
    
    # 收集主文件夹中的图片文件哈希值，用于去重
    main_folder_hashes = {}
    main_image_files = []
    
    # 找出主文件夹中所有图片文件并计算哈希值
    for file in os.listdir(main_folder):
        file_path = os.path.join(main_folder, file)
        if os.path.isfile(file_path):
            # 检查文件扩展名是否为图片格式
            ext = os.path.splitext(file)[1].lower()
            if ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']:
                main_image_files.append(file_path)
                main_folder_hashes[get_file_hash(file_path)] = file_path
    
    # 准备重命名子文件夹中的图片，从6290开始
    rename_counter = 6290
    
    # 处理子文件夹中的图片
    processed_files = []
    duplicate_files = []
    
    for file in os.listdir(sub_folder):
        file_path = os.path.join(sub_folder, file)
        if os.path.isfile(file_path):
            ext = os.path.splitext(file)[1].lower()
            if ext in ['.jpg', '.jpeg', '.png', '.gif', '.bmp']:
                # 计算文件哈希值，检查是否重复
                file_hash = get_file_hash(file_path)
                
                if file_hash in main_folder_hashes:
                    # 发现重复文件，加入删除列表
                    duplicate_files.append(file_path)
                    print(f"发现重复文件，将删除: {file_path}")
                    continue
                
                # 为文件生成新名称，格式为0006290.jpg
                new_file_name = f"{rename_counter:07d}.jpg"
                new_file_path = os.path.join(main_folder, new_file_name)
                
                # 如果文件已经是JPG/JPEG格式且不需要转换，直接复制
                if ext in ['.jpg', '.jpeg']:
                    shutil.copy2(file_path, new_file_path)
                else:
                    # 转换为JPEG格式
                    if convert_to_jpeg(file_path, new_file_path):
                        processed_files.append(new_file_path)
                        
                rename_counter += 1
    
    # 删除子文件夹中的重复文件
    for file_path in duplicate_files:
        try:
            os.remove(file_path)
        except Exception as e:
            print(f"删除重复文件失败 {file_path}: {e}")
    
    # 删除子文件夹
    try:
        shutil.rmtree(sub_folder)
        print(f"已删除子文件夹: {sub_folder}")
    except Exception as e:
        print(f"删除子文件夹失败: {e}")
    
    # 统计信息
    print(f"\n处理完成!")
    print(f"主文件夹中的图片数量: {len(main_image_files)}")
    print(f"子文件夹中处理的图片数量: {len(processed_files)}")
    print(f"发现并删除的重复图片数量: {len(duplicate_files)}")

if __name__ == "__main__":
    main()