- 博客(2)
- 收藏
- 关注
RSS订阅原创 跑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关注的人