- 博客(13)
- 收藏
- 关注
原创 配置pytorch_geometric
pytorch_geometric是一个图神经网络的库。如果直接通过pip安装在运行时会显示依赖库(如torch-sparse)没安装。在安装这些依赖库通过pip安装不上。需要去官网 https://pytorch-geometric.com/whl/ 下载对应pytorch和cuda版本的依赖库。或者用这样的代码进行安装(pytorch和cuda版本需要修改)...
2022-06-22 09:54:43
402
1
原创 HigherHRnet/DEKR骨骼关键点检测笔记:heatmap和offset理解
本文是对HigherHRnet/DEKR骨骼关键点检测项目代码阅读后的笔记。Github:https://github.com/HRNet/DEKR论文:https://arxiv.org/abs/2104.02300HigherHRnet/DEKR骨骼关键点检测是MIT提出一种新颖的bottom-up方法,并且在COCO和Crowdpose中达到了SOTA,利用HRnet网络预测骨骼点的heatmap和offset,得到骨骼关键点的坐标和置信度。heatmap+offset是谷歌在CVPR2017
2021-07-09 10:29:33
4911
3
原创 解决问题:Do not know how to handle these types to promote: {‘DoubleTensor‘, ‘FloatTensor‘}
今天在使用pytorch1.7训练模型,调用apex时出现了该问题:Doubletensor是64位float类型,Floattensor是32位float类型,调试程序发现第二次迭代时,变量v和msd[k]的数据类型不同,如图所示:因此在条件循环后添加程序: if v.dtype != msd[k].dtype: v = v.to(msd[k].dtype)修改后如图:解决问题!...
2021-03-01 17:43:18
1420
原创 原神甘雨伤害计算程序 【python】
calc_damage可以计算普通人物的伤害期望,calc_damage_Ganyu主要针对重击伤害和技能伤害,这里没有考虑冻结时额外加的20暴击率。别问,问就是没水。import numpy as npdef calc_damage(base_attack, ext_attack, critical, critical_damage): return (base_attack + ext_attack) * critical * (1 + critical_damage)\
2021-01-27 09:14:23
2415
原创 Faster-RCNN结构原理图
看了很久Faster-RCNN的代码论文博客,总算把它的结构和原理搞明白了,为了加深记忆和帮助其他人学习,画了一个Faster-RCNN结构原理图,只是一个较为大致的框架原理,其中的具体细节还是得自己去看看代码才能明白。主要参考:知乎文章《一文读懂Faster RCNN》:https://zhuanlan.zhihu.com/p/31426458参考代码为:https://github.com/chenyuntc/simple-faster-rcnn-pytorch...
2021-01-23 15:12:11
849
原创 pytorch模型运算时间的不同
今天测试模型计算时间时,运行了这样一段代码:在调用时发现 首次调用第一个模型的计算时间 会比之后的模型计算时间长很多,运行结果如图:可以看到即便是同样是resnet50,第一次调用和第二次调用时间差别非常大。而且即便我改变了模型加载的顺序,在调用时仍然是第一个模型的计算时间远超其他模型。因此我判断python程序在调用pytorch框架下第一个模型计算时,运算时间会比之后再计算时间长很多,并且用循环体做了另一个实验,我运行了这样一段程序:可以看到即使是同样的模型,第一次调用它计算的所耗
2021-01-06 11:37:01
1465
3
原创 win10配置detetron2踩过的坑
主要是参考该博客配置的:https://blog.csdn.net/weixin_43448140/article/details/109397059但是配置过程中遇到一些问题:1.我是用anaconda配置虚拟环境,再在虚拟环境下跑的,但是系统的cuda环境也会在安装中调用到,因此必须系统的cuda环境也配置为10.2,不然在运行detetron2的setup.py时会报错。2.用mingw配置gcc时需要翻墙,如果没法翻可以参考https://blog.csdn.net/qq_34432960/a
2020-11-25 15:35:24
318
原创 直接解决OMP: Error #15: Initializing libiomp5md.dll, but found libiomp5md.dll already initialized.报错问题
今天想通过cmd安装github上的python包,遇到了这个问题:以前在pycharm里调试程序时可以直接通过在程序前添加这两个语句解决import osos.environ[“KMP_DUPLICATE_LIB_OK”]=“TRUE”但在这种情况并不能解决问题。甚至直接在terminal上import torch也会出现这种问题:究其原因其实是,anaconda的环境下存在两个libiomp5md.dll文件。所以直接去虚拟环境的路径下搜索这个文件,可以看到在环境里有两个dll文件:
2020-11-24 15:16:52
39763
45
原创 Python opencv在图像上画XML对应的框
主要用于显示数据集的标注是否正确。只写了显示的部分,并未保存图片,而且由于只有一个分类,并未在框上写明类别import osimport xml.dom.minidomimport cv2import xml.etree.ElementTree as ETclasses = ["phone"] # XML文件中的所有类别ImgPath = 'D:/Wei X/data/phone image/' # 图片文件夹路径AnnoPath = 'D:/Wei X/data/phone xml
2020-10-23 10:21:06
699
转载 【转】利用setuptools打包python程序
https://www.cnblogs.com/leokale-zz/p/12207923.html
2020-09-28 20:08:56
111
原创 Ubuntu18.04 Anaconda环境下配置pytorch(gpu)+tensorboardX
之前在网上看了很多教程,由于好多都过时了,存在不少弯路和坑,因此写一篇比较简便的配置方法。本文参考了tensorboardX的官方文档,https://github.com/lanpa/tensorboardX可以看到目前的tensorboardX再以下环境测试过:*The current release (v2.1) is tested on anaconda3, with PyTorch 1.5.1/ torchvision 0.6.1 / tensorboard 2.2.2.目标
2020-09-11 11:11:08
859
原创 基于anaconda和python3.7安装pytorch1.5.0
一开始看到自己电脑的CUDA版本为11.0,以为不能支持,实际上在虚拟环境中可以安装cudatoolkit的任意版本:1.打开anaconda prompt,建立一个新环境conda create -n your_env_name python=3.72.安装cudatoolkit和对应的cudnnconda install cudatoolkit=10.2conda install cudnn=7.6.53.安装pytorch直接去官网查找指令即可https://pytorch.org/我
2020-07-03 17:02:05
1068
2
原创 python配置opencv+contrib4.1.2教程
python配置opencv+contrib4.1.2教程今天安装pycharm和anaconda后配置了很久opencv和contrib,发现网上很多的方法都不好使。所以来记一个自己能用的配置方法:使用条件:python版本3.7已经安装好anaconda和pycharm,添加anaconda到环境变量中,并且创立了一个自己的环境下载opencv4.1.2和对应的contrib下载地...
2020-04-09 19:19:53
2077
6
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人