- 博客(2)
- 收藏
- 关注
原创 跑unet替换resnet50时出现的两个报错:Attempted to set the storage of a tensor on device “cpu“ to a storage
对于我这个菜鸡来说,这个真的改了好久好久,在网上搜,有的说是版本不匹配,最后还把自己环境搞的不能用了。ps:记得把unet.py权重文件换成训练完之后的best_epoch_weights.pth,再运行get_miou.py。网上直接下载的Unet,里面可以选择使用哪种骨干网络,vgg直接运行没毛病,现在换成resnet50运行就报错了。运行get_miou.py文件又出现了新的报错,刚开始的解决办法不管用了。先训练,运行train.py刚开始的报错如下:(忘记截图了)改的是文件unet.py里面。
2024-06-12 16:38:42
254
原创 跑图片分割unet把无人机采集的大图裁剪为小图识别
'D:/数据集/自选数据集/无人机大图/水平1.png', 24064, 4096) #原本图片是23818x4027,改成了24064x4096。save_images(image_list, 'D:/数据集/自选数据集/无人机大图/img_小图') #裁成小图的文件名。add_white_edge('D:/数据集/自选数据集/无人机大图/水平.png', #原大图路径。file_path = "D:/数据集/自选数据集/无人机大图/水平1.png" #原文件。
2024-05-24 10:43:13
415
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人