HAN&DAI

  • 首页
  • 遥感应用
  • GIS应用
  • 机器学习
  • 实用工具
  • 文章链接
  • 遥感数据集
pytorch
pytorch使用

减少命令行长度并简化配置管理

问题:执行以下命令行太长了,不方便修改。 CUDA_VISIBLE_DEVICES=0 python tools/train.py --config ./configs/SegSAMPLerEmbMLP_config.py --cfg-options optimizer.lr=0.0005 optimizer.weight_decay=0.01 train_batch_size_per_gpu=4 test_batch_size_per_gpu=4 model_cfg.SAM_weights=0.1 model_c…

2024年11月12日 0条评论 186点热度 1人点赞 daidai 阅读全文
BUG修改

PyTorch转ONNX踩坑——RuntimeError: ONNX export of operator adaptive_avg_pool2d & Failed to export an ONNX attribute 'onnx::Gather'

踩坑一: RuntimeError: ONNX export of operator adaptive_avg_pool2d 报错原因:这个错误表明在将模型导出为 ONNX 格式时,adaptive_avg_pool2d 操作不支持当前的输出大小,因为输出大小不是输入大小的因数。因为onnx要求模型的输入参数的固定的,而自适应池化层是根据输入来确定模型参数的。 ChatGPT提供了一些别的方法,如更新Pytorch版本或尝试使用 ONNX opset 版本或 dynamic_axes 参数,经过测试都无法解决以上问…

2024年11月8日 0条评论 469点热度 1人点赞 daidai 阅读全文
pytorch使用

Pytorch图像分割模型转ONNX

最近学习了一下图像分割模型部署——PyTorch转ONNX 参考了子豪兄的视频 一、主要学习内容如下: 即ONNX是个中间翻译器,帮助不同框架和不同设备更方便地迁移和部署。 二、具体操作 1. 安装配置环境 安装Pytorch pip3 install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu113 安装 ONNX pip install onnx -i https://pypi.tuna.tsinghua.edu…

2024年11月8日 0条评论 225点热度 1人点赞 daidai 阅读全文
遥感应用

mmrotate旋转框目标检测框架部署

提出问题 当前,有众多的开源方法集合以及社区维护。为了满足旋转框目标检测需求,一些学者在pytorch深度学习框架基础上构建了mmrotate旋转框目标检测体系。借此机会,自己搭建了一下mmrotate的环境,加深学习和为后续做相关对比实验做一些准备工作。 解决方案 一台2080或者2080ti的机器,cuda 10.2 参考代码: https://github.com/open-mmlab/mmrotate/ 环境部署步骤 首先要下载两个代码和相关库,链接分别为:mmrotate、mmdetection和mmcv…

2024年4月1日 0条评论 490点热度 1人点赞 HAN&DAI 阅读全文
实用工具

Win7下docker环境配置相关记录

提出问题 因为项目环境部署需要,需要在Win7环境下部署一个算法,而且是在无网情况下。因此,考虑使用docker来执行这次部署任务,然而也不是十分顺利,所以来简单记录一下配置流程。 参考链接:Windows 7 安装Docker实践 解决方案 因为担心docker版本较高不能在win7系统上运行,因此,我们安装了一个DockerToolbox-19.03.1版本。总体来说,win7版本的docker和win10不太一样,安装步骤可以按照上述参考链接 Windows 7 安装Docker实践 去安装。安装之后,桌面会…

2023年11月2日 0条评论 385点热度 2人点赞 HAN&DAI 阅读全文
pytorch使用

在docker中使用tensorboard、jupyter notebook

由于在docker中创建容器的时候已经使用了默认的端口映射,因此,再想在容器中使用tensorboard或者jupyter的时候,会出现无法打卡网页的情况。所以需要在创建容器的时候分配好新的端口,从而实现docker容器与宿主机之间的映射。 一、创建容器: docker run --gpus all --shm-size 16G --name dete2 -it -p 8888:8888 -p 6006:6006 -v /home/daidai/bigdata:/home/daidai/bigdata d91fe8…

2023年4月13日 0条评论 756点热度 1人点赞 daidai 阅读全文
BUG修改

BUG: torch.pairwise_distance()计算欧式距离报错,出现size不匹配的情况

由于网络中用到了torch.pairwise_distance(),发现在不同版本的pytorch下,计算结果不一致的情况。下面就来解决这个问题。 一、torch.nn. PairwiseDistance ( p = 2.0 , eps = 1e-06 , keepdim = False )用法 首先,了解一下pytorch里 torch.pairwise_distance()的用法。 其中,各项参数如下: p ( real , optional ) – 范数度。可以为负。默认值:2 eps ( float , o…

2023年3月31日 0条评论 546点热度 1人点赞 daidai 阅读全文

HAN&DAI

RS和GIS研究兴趣者,永远在学习的路上!

网页统计
   

本机访问量:0

欢迎交流,QQ群:821388027

最新 热点 随机
最新 热点 随机
减少命令行长度并简化配置管理 PyTorch转ONNX踩坑——RuntimeError: ONNX export of operator adaptive_avg_pool2d & Failed to export an ONNX attribute 'onnx::Gather' Pytorch图像分割模型转ONNX 边缘端部署智能化图像处理算法-目标检测与跟踪系列 科研干货分享 mmdetection和mmtracking结合下的目标检测与跟踪(包含数据集制作)
遥感影像/自然影像手动点选距离(长度)计算 大尺度遥感线性地物(道路)提取后细化(骨架提取) 遥感专用深度学习框架LuoJiaNet系列学习(1)---源码安装 目标跟踪数据集格式转换(按帧排序->按目标轨迹ID排序) 给定语句提取含有地理信息的省份-第一步 labelme等标注软件多分类(二分类)json文件转mask(可rgb显示或one-hot显示)
浏览最多的文章
  • BUG:ImportError: /lib/x86_64-linux-gnu/libstdc++.so.6: version `GLIBCXX_3.4.29' not found (1,501)
  • 利用GEE下载指定区域Landsat8影像 (1,288)
  • BUG:“ModuleNotFoundError: No module named '_ext'”的解决方案 (1,285)
  • 利用arcgis制作深度学习标签数据(以二分类为例) (948)
  • 利用传统机器学习方法进行遥感影像分类-以随机森林(RF)为例 (865)
专题分类
标签聚合
遥感影像应用 pytorch GIS应用 实用技巧 机器学习 环境配置
归档博文
  • 2024 年 11 月 / 3篇
  • 2024 年 8 月 / 2篇
  • 2024 年 6 月 / 1篇
  • 2024 年 5 月 / 4篇
  • 2024 年 4 月 / 1篇
  • 2023 年 12 月 / 1篇
  • 2023 年 11 月 / 2篇
  • 2023 年 9 月 / 2篇
  • 2023 年 8 月 / 3篇
  • 2023 年 7 月 / 6篇
  • 2023 年 6 月 / 1篇
  • 2023 年 5 月 / 5篇
  • 2023 年 4 月 / 3篇
  • 2023 年 3 月 / 1篇
  • 2023 年 2 月 / 4篇
  • 2022 年 12 月 / 1篇
  • 2022 年 9 月 / 3篇
  • 2022 年 8 月 / 5篇
  • 2022 年 7 月 / 1篇
  • 2022 年 6 月 / 1篇
  • 2022 年 5 月 / 2篇
  • 2022 年 4 月 / 7篇

COPYRIGHT © 2025 HAN&DAI. ALL RIGHTS RESERVED. QQ交流群:821388027

Theme Kratos Made By Seaton Jiang