【论文10】复现代码tips

一、准备工作

1.创建一个虚拟环境

conda create --name drgcnn38 python=3.8.18

2.激活虚拟环境

conda activate drgcnn38

注意事项

在Pycharm中终端（terminal）显示PS而不是虚拟环境base

问题如下所示

解决方法：shell路径改成cmd.exe

重启终端显示虚拟环境

3.安装torch 

conda install pytorch==2.1.0 torchvision==0.16.0 torchaudio==2.1.0 cpuonly -c pytorch

安装一系列包

注意事项

Pycharm远程连接Linux服务器实现代码同步

1.工具-->部署-->配置

2.选择SFTP远程连接，路径填与服务器要同步的路径地址

二、代码学习

各部分的作用

• eye_pre_process：视网膜眼底图像预处理模块。
• Encoder：编码器训练模块。
• modules：包含模型结构、损失函数和学习率降低策略。
• utils：包含一些常用函数和评估指标。
• BFFN：双眼特征融合网络训练模块。
• CAM：类别注意力模块。

eye_pre_process

copy.py

# 创建一个ArgumentParser对象，用于处理命令行参数  
parser = argparse.ArgumentParser()  # 添加一个命令行参数 '--image-folder'，类型为字符串，默认值为 'D:/cv_paper/lesson/Dataset/ceshi'  
# 这个参数用于指定输入图像的文件夹路径  
parser.add_argument('--image-folder', type=str, default=r'D:/cv_paper/lesson/Dataset/ceshi')  # 添加一个命令行参数 '--output-folder'，类型为字符串，默认值为 'D:\cv_paper\lesson/Dataset/ceshi_output'  
# 注意：这里路径中的反斜杠在不同的操作系统中可能需要特别注意，Python字符串中推荐使用原始字符串（r前缀）来避免转义字符的问题  
# 这个参数用于指定输出结果的文件夹路径  
parser.add_argument('--output-folder', type=str, default=r'D:\cv_paper\lesson/Dataset/ceshi_output')  # 添加一个命令行参数 '--crop-size'，类型为整数，默认值为512  
# 这个参数用于指定图像裁剪的大小  
parser.add_argument('--crop-size', type=int, default=512, help='crop size of image')  # 添加一个命令行参数 '-n' 或 '--num-processes'，类型为整数，默认值为8  
# 这个参数用于指定处理任务时要使用的进程数  
# '-n' 是 '--num-processes' 的简写形式，帮助信息说明了该参数的作用  
parser.add_argument('-n', '--num-processes', type=int, default=8, help='number of processes to use')

# 转换一个包含多个任务的列表，每个任务由文件名、目标路径和裁剪大小组成  
# 对于jobs列表中的每个任务（索引为j），它首先检查是否已经处理了100个任务（作为进度指示），然后调用convert函数来执行实际的图像转换。
def convert_list(i, jobs):  for j, job in enumerate(jobs):  # 每处理100个任务打印一次进度  if j % 100 == 0:  print(f'worker{i} has finished {j} tasks.')  # 解包任务元组并调用convert函数  convert(*job)  # 转换单个图像文件，包括模糊处理、裁剪和保存  
def convert(fname, tgt_path, crop_size):  img = Image.open(fname)  # 打开图像文件  blurred = img.filter(ImageFilter.BLUR)  # 应用模糊滤镜  ba = np.array(blurred)  # 将图像转换为NumPy数组  h, w, _ = ba.shape  # 获取图像的高度、宽度和通道数  # 尝试根据图像的亮度分布来识别前景区域  if w > 1.2 * h:  # 计算左右两侧的最大亮度值  left_max = ba[:, :w // 32, :].max(axis=(0, 1)).astype(int)  right_max = ba[:, -w // 32:, :].max(axis=(0, 1)).astype(int)  max_bg = np.maximum(left_max, right_max)  foreground = (ba > max_bg + 10).astype(np.uint8)  # 识别前景区域  bbox = Image.fromarray(foreground).getbbox()  # 获取前景区域的最小边界框  # 如果边界框太小或不存在，则打印消息并可能设置为None  if bbox is None:  print(f'No bounding box found for {fname} (???)')  else:  left, upper, right, lower = bbox  if right - left < 0.8 * h or lower - upper < 0.8 * h:  print(f'Bounding box too small for {fname}')  bbox = None  else:  bbox = None  # 如果图像已经是合适的宽高比，则不尝试识别前景  # 如果未找到有效的边界框，则使用正方形边界框  if bbox is None:  bbox = square_bbox(img)  # 使用边界框裁剪图像，并调整大小  cropped = img.crop(bbox)  cropped = cropped.resize([crop_size, crop_size], Image.ANTIALIAS)  # 注意：ANTIALIAS可能是个拼写错误，应该是ANTIALIASIS  save(cropped, tgt_path)  # 保存图像  # 返回一个正方形裁剪框的边界  
def square_bbox(img):  w, h = img.size  left = max((w - h) // 2, 0)  upper = 0  right = min(w - (w - h) // 2, w)  lower = h  return (left, upper, right, lower)  # 保存PIL图像到文件  
def save(img, fname):  img.save(fname, quality=100, subsampling=0)  # 注意：subsampling参数可能不是所有格式都支持  # 假设的main函数，用于组织整个流程（注意：这里只是一个示例）  
def main():  # 示例任务列表，每个任务是一个(文件名, 目标路径, 裁剪大小)元组  jobs = [  ('input1.jpg', 'output1_resized.jpg', 256),  ('input2.jpg', 'output2_resized.jpg', 256),  # ... 更多任务  ]  # 假设有一个工作者ID为1  convert_list(1, jobs)  if __name__ == "__main__":  main()  

Encoder 

main.py

# 定义主函数入口  
def main():  # 解析配置参数  args = parse_configuration()  # 加载配置文件  cfg = load_config(args.config)  # 获取配置中保存的路径  save_path = cfg.config_base.config_save_path  # 如果保存路径不存在，则创建该路径  if not os.path.exists(save_path):  os.makedirs(save_path)  # 将配置文件复制到保存路径  copy_config(args.config, cfg.config_base.config_save_path)  # 执行工作函数  worker(cfg)  # 定义工作函数，负责训练、验证和测试模型  
def worker(cfg):  # 根据配置生成模型  model = generate_model(cfg)  # 计算模型总参数数量  total_param = 0  for param in model.parameters():  total_param += param.numel()  print("Parameter: %.2fM" % (total_param / 1e6))  # 打印模型参数数量（单位：百万）  # 根据配置生成训练、验证和测试数据集  train_dataset, test_dataset, val_dataset = generate_dataset(cfg)  # 初始化性能评估器  estimator = PerformanceEvaluator(cfg.config_train.config_criterion, cfg.config_data.config_num_classes)  # 执行训练过程  train(  cfg=cfg,  model=model,  train_dataset=train_dataset,  val_dataset=val_dataset,  estimator=estimator,  )  # 测试最佳验证模型性能  print('This is the performance of the best validation model:')  checkpoint = os.path.join(cfg.config_base.config_save_path, 'best_validation_weights.pt')  cfg.config_train.config_checkpoint = checkpoint  # 设置检查点路径为最佳验证模型  model = generate_model(cfg)  # 重新生成模型以加载权重  evaluate(cfg, model, test_dataset, estimator)  # 评估模型性能  # 测试最终模型性能  print('This is the performance of the final model:')  checkpoint = os.path.join(cfg.config_base.config_save_path, 'final_weights.pt')  cfg.config_train.config_checkpoint = checkpoint  # 设置检查点路径为最终模型  model = generate_model(cfg)  # 重新生成模型以加载权重  evaluate(cfg, model, test_dataset, estimator)  # 评估模型性能  # 如果此脚本作为主程序运行，则调用main函数  
if __name__ == '__main__':  main()

Encoder_predict.py

进行模型的训练，具体来说，它定义了一个训练循环&#x