SAM大模型实践(一)

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

参考着segment-geospatial 项目主页的介绍,尝试复现一下Example-satallite的案例。

  Satellite - segment-geospatialicon-default.png?t=O83Ahttps://samgeo.gishub.org/examples/satellite/         过程当中遇到了一些坑给大家做点分享,主要有几种情况,一个是torch、torchvision、torchaudio和cuda安装的问题,版本不一致,运行过程中报错NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend. 这个时候参考了其他博主的解决方法,将torch、torchvision、torchaudio 卸载重装了一致的版本,如果你的电脑上面有GPU尽量安装GPU版本。【问题解决】NotImplementedError: Could not run ‘torchvision::nms‘ with arguments from the ‘CUDA‘ backend._notimplementederror: could not run 'torchvision::n-优快云博客文章浏览阅读1k次,点赞6次,收藏5次。本文描述了关于Pytorch的报错“NotImplementedError: Could not run 'torchvision::nms' with arguments from the 'CUDA' backend.”并提供了解决方案。_notimplementederror: could not run 'torchvision::nms' with arguments from thhttps://blog.youkuaiyun.com/m0_55213577/article/details/143097871

全网最详细的安装pytorch GPU方法,一次安装成功!!包括安装失败后的处理方法!-优快云博客文章浏览阅读10w+次,点赞179次,收藏769次。PyTorch的GPU版本利用了NVIDIA的CUDA技术,使得深度学习计算能够高效地在GPU上运行。使用GPU来执行深度学习计算可以显著加速计算,从而减少训练和推理时间。CUDA是NVIDIA推出的一种通用并行计算架构,可以使GPU执行通用计算任务,而不仅仅是图形处理。在PyTorch中,可以使用CUDA来利用NVIDIA GPU的并行计算能力加速模型训练和推理。cuDNN是NVIDIA专门为深度学习模型设计的一个库,它提供了高效的卷积操作和其他计算操作,可以进一步加速深度学习任务。_全网最详细的安装pytorch gpuhttps://blog.youkuaiyun.com/qlkaicx/article/details/134577555      另外一个情况是leafmap用jupter notebook没有跑成功,可能是依赖包的问题。但是在colab里面跑成功了,地图和图层都正常显示了。在jupter notebook显示不出来没关系,不影响后续SAM模型的应用。

按照流程一步步跑下去,最后的一些结果如下所示:

下载一幅11000×14000的遥感影像

分割过程花了大概一个小时左右,我的电脑配置是thinkbook 16p,带有8G的GPU,时间还是比较久的。

最后的分割效果如下:

您可能感兴趣的与本文相关的镜像

PyTorch 2.5

PyTorch 2.5

PyTorch
Cuda

PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

### 关于 SAM 大模型的学习路径 学习 SAMSegment Anything Model)大模型需要掌握多个方面的知识,包括但不限于计算机视觉的基础理论、深度学习框架的应用以及具体到 SAM 的实现细节。以下是关于 SAM 大模型的学习路径: #### 1. 计算机视觉基础知识 理解 SAM 需要具备扎实的计算机视觉基础,尤其是语义分割和实例分割的相关概念。可以参考经典的教材或在线课程来学习这些内容。 - 推荐书籍:《Computer Vision: Algorithms and Applications》[^5]。 - 在线资源:Coursera 上的 Computer Vision 特专项课程[^6]。 #### 2. 深度学习框架应用 SAM 是基于 PyTorch 实现的大规模模型,因此熟悉 PyTorch 对理解和实践 SAM 至关重要。 - 官方文档:PyTorch 提供了详尽的入门指南和高级教程[^7]。 - 示例代码: ```python import torch from torchvision.models.segmentation import fcn_resnet50 model = fcn_resnet50(pretrained=True) print(model) ``` #### 3. SAM 的安装与使用 根据引用描述,在 Google Colab 平台上运行 SAM 算法相对简单。可以通过以下步骤快速上手 SAM 地理空间版本的安装和基本操作[^1]。 - 安装依赖项: ```bash pip install -U segment-geospatial ``` - 导入必要库并初始化模型: ```python import leafmap from samgeo import SamGeo, regularize sam_model = SamGeo() ``` #### 4. 数据处理与预训练模型选择 为了获得更好的效果,需注意数据格式化的过程,例如波段顺序调整和数据类型的转换。此外,选择合适的预训练模型也是关键步[^2]。 #### 5. 性能评估方法 在完成模型部署之前,应对其进行全面测试以确保其稳定性和准确性。这涉及多维度的评价标准和技术手段[^3]。 #### 6. 扩展阅读与研究方向 随着模型尺寸的增长,相应的计算需求也会显著提高;然而,只有当训练集大小同步增长时才能持续改善性能[^4]。对此现象的研究有助于优化未来的工作流程。 --- ###
评论
成就一亿技术人!
拼手气红包6.0元
还能输入1000个字符
 
红包 添加红包
表情包 插入表情
 条评论被折叠 查看
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值