使用PASCAL VOC 2012数据集进行标准化处理,以适应网络输入,并且将PASCAL VOC 2012和SDB数据集混合,进行数据增强
编写了自定义的SegDataset
类,用于加载和预处理数据,包括数据增强和格式化
导入VGG-16预训练权重:使用预训练权重初始化网络,加速收敛。
并且从包括数据下载、数据预处理、网络构建、训练、评估和推理等各个环节,都有实际的代码和注释,很方便直观
使用PASCAL VOC 2012数据集进行标准化处理,以适应网络输入,并且将PASCAL VOC 2012和SDB数据集混合,进行数据增强
编写了自定义的SegDataset
类,用于加载和预处理数据,包括数据增强和格式化
导入VGG-16预训练权重:使用预训练权重初始化网络,加速收敛。
并且从包括数据下载、数据预处理、网络构建、训练、评估和推理等各个环节,都有实际的代码和注释,很方便直观