- 博客(2)
- 收藏
- 关注
原创 UNet复现过程中遇到的问题以及解决
它可以帮助您在训练模型时记录实验结果、比较不同参数设置的效果,并与团队成员共享实验结果。在train.py文件中找到下图三个路径分别改成自己的训练数据集位置、标签集位置以及生成训练模型的位置。在最近的学习中想要用UNet网络进行建筑物的分割检测,于是从github上下载了网络,尝试进行复现。碰到的问题,在这记录一下。接下来去寻找通道数,如果自己的数据集是RGB图像则设置n_channel为3,黑白图像则为1。-i后改成预测输入图像 -o 改成要保存的路径 第二行代码是输出图像不保存版本的。
2023-11-15 21:22:35
272
原创 wandb在pycharm中安装与配置经验分享
Wandb (Weights & Biases) 是一个用于跟踪、可视化和协作机器学习实验的平台。它可以帮助您在训练模型时记录实验结果、比较不同参数设置的效果,并与团队成员共享实验结果。安装环境:anaconda下pytorch环境中。将你的API直接粘贴到password上即可。将自己的API复制粘贴到第三行,这里。在c盘-用户中查找.netrc文件。注意粘贴过来的API可能没有显示。可以在第二行查看自己的API。回到terminal输入。当出现下图时则安装成功。其中第一行为帮助文档。
2023-11-13 15:40:47
1455
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人