- 博客(2)
- 收藏
- 关注
RSS订阅原创 【Pytorch-UNet】train.py
random_split()函数随机划分数据集，dataset-原始数据集，[n_train,n_val]是一个列表，指定每个子集的样本数量。generator=torch.Generator().manual_seed(0)：创建一个固定随机种子的生成器，保证每次运行代码时划分结果一致。生成器的种子为0（种子可以是任意整数），相同种子→相同随机序列→相同数据划分结果。
2025-05-29 21:13:39
705
原创 【Pytorch-Unet】细胞分割问题
1. 修改get_output_filenames函数，生成以原文件名加上“_mask.tif”的输出文件名。3. 处理输入参数为目录的情况，遍历目录下的所有图片文件，并为每个文件生成对应的输出路径。使用的数据集为灰度图，channels=1（如果是RGB，则channels=3）。2. 确保输出路径正确，包括将输出文件保存到用户指定的目录，如果目录不存在则创建。问题四：测试图像保存路径出错，输出文件名为字母而不是测试图片名称。遍历输入目录中的图片文件，生成对应的输出文件名（例如。
2025-04-07 21:21:43
372
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人