MSPFN 代码复现

原创

已于 2025-02-09 13:51:20 修改 · 367 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#深度学习 #人工智能 #机器学习 #笔记 #学习

于 2025-02-07 00:08:41 首次发布

1、环境配置

conda create -n MSPFN python=3.9

conda activate MSPFN

pip install opencv-python

pip install tensorflow

pip install tqdm

pip install matplotlib

2、train

2.1 创建数据集

2.1.1 数据集格式

|--rainysamples  
    |--file1
            ：  
    |--file2
        :
    |--filen
    
|--clean samples
    |--file1
            ：  
    |--file2
        :
    |--filen

2.1.2 将图片1.jpg 放到file1 的代码

import os
import shutil

# 源文件夹路径，存放图片的文件夹
source_folder = r'path/to/your/source/folder'

# 获取源文件夹中的所有文件
files = os.listdir(source_folder)

# 遍历文件
for file_name in files:
    # 检查文件是否为.jpg格式
    if file_name.endswith('.jpg'):
        # 提取文件名中的编号（去掉扩展名）
        file_number = os.path.splitext(file_name)[0]
        
        # 创建目标文件夹路径
        target_folder = os.path.join(source_folder,