深度学习
文章平均质量分 63
知识在于分享
即便是蜗牛,一步一步不停歇的爬也会有看见太阳的一天
展开
-
pytorch查看显存利用率
如何在Pytorch中精细化利用显存 - Oldpan的个人博客再次浅谈Pytorch中的显存利用问题(附完善显存跟踪代码) - Oldpan的个人博客浅谈深度学习:如何计算模型以及中间变量的显存占用大小 - Oldpan的个人博客...转载 2021-10-16 17:35:05 · 1156 阅读 · 0 评论 -
tensorflow1.14 转换pytorch 部署inference 所遇到问题 superpoint
GitHub - rpautrat/SuperPoint: Efficient neural feature detector and descriptor1.checkpoint转换pbSuperPoint/export_model.py at master · rpautrat/SuperPoint · GitHub2.freeze_graph参考1:Python freeze_graph.freeze_graph方法代碼示例Python freeze_graph.freeze_原创 2021-10-15 14:47:26 · 834 阅读 · 0 评论 -
JupyterLab-APP
https://github.com/jupyterlab/jupyterlab_app/mac版本windows版本liunx版本注意下载的时候看一下该版本中JupyterLab-APP,的JupyterLab的版本比如:JupyterLab-APP绑定的是JupyterLab v3.1.13这个软件自带python+conda环境ubuntu默认安装在 /opt/JupyterLab/resources/jlab_server/如果你想对这个环境安装其他东西..原创 2021-09-30 13:43:07 · 1224 阅读 · 0 评论 -
superpoint的onnx转换全流程
完整的SSD-tf版本转换为onnx示例:https://github.com/onnx/tensorflow-onnx/blob/master/tutorials/ConvertingSSDMobilenetToONNX.ipynb本文所示示例为superpointhttps://github.com/rpautrat/SuperPointmagic-point网络训练模拟数据input(h,w,c)=(120,160,1)magic-point网络导出coco数据的GT,使用input(原创 2021-09-29 15:08:26 · 1851 阅读 · 0 评论 -
onnx-onnxruntime的一些官方文档记录
# onnxruntime-gpu与CUDA版本对应关系# https://onnxruntime.ai/docs/execution-providers/CUDA-ExecutionProvider.html# cuda10.1 cudnn7.6.5 onnxruntime-gpu-1.4.0# onnx现有算子对应的OpSet Version# https://github.com/microsoft/onnxruntime/blob/v1.4.0/docs/OperatorKern原创 2021-09-27 13:42:32 · 864 阅读 · 0 评论 -
mmdet-tools工具测试
analyze_logs.pypython analyze_logs.py plot_curve log.json --keys loss_cls --legend loss_clspython analyze_logs.py plot_curve log.json --keys loss_cls loss_bbox --legend loss_cls loss_bboxpython analyze_logs.py plot_curve log.json --keys lo...原创 2021-09-16 12:43:13 · 2809 阅读 · 7 评论 -
tensorflow-v1.12的保存模型方式学习
tensorflow保存模型的几种方法tensorflow保存模型的几种方法_zmlovelx(帅得不敢出门 程序员群31843264)-CSDN博客tensorflow中的检查点checkpoint详解(二)——以tensorflow1.x 的模型保存与恢复为主tensorflow中的检查点checkpoint详解(二)——以tensorflow1.x 的模型保存与恢复为主_MIss-Y的博客-CSDN博客tensorflow模型保存(三)——tensorflow1.x版本的savedmode原创 2021-09-15 02:05:23 · 462 阅读 · 0 评论 -
transformer中softmax为什么除根号d_k
输入值不能太大,太大如下,梯度消失import torchfrom torch import nnfrom torch.nn import functional as Fimport numpy as np#from torch.autograd import Variable#from torchvision.models import resnet50#import torchvision.transforms as T#torch.set_grad_enabled(True).原创 2021-09-08 15:32:06 · 1739 阅读 · 0 评论 -
transformer资料
几个博客:Transformer 详解https://wmathor.com/index.php/archives/1438/英文博客https://jalammar.github.io/illustrated-transformer/中文翻译:可视化理解Transformer结构https://zhuanlan.zhihu.com/p/59629215https://blog.csdn.net/longxinchen_ml/article/details/86533005一些可视化原创 2021-09-08 10:55:03 · 207 阅读 · 0 评论 -
转置卷积的一些思考
#参考链接#一文搞懂 deconvolution、transposed convolution、sub-pixel or fractional convolution#https://www.cnblogs.com/shine-lee/p/11559825.html#ConvTranspose2d原理,深度网络如何进行上采样?#https://blog.csdn.net/qq_27261889/article/details/86304061#Pytorch: conv2d、空洞卷积、m原创 2021-08-30 23:12:28 · 203 阅读 · 1 评论 -
coco格式的json做离线数据增强,随机把图像bbox数据crop贴图,并移除对应的ann
import cv2import osimport numpy as npimport jsonimport randomfrom pycocotools.coco import COCO, maskUtilsROOT = os.path.dirname(os.path.abspath(__file__))BASE = os.path.dirname(ROOT)def mkdir_os(path): if not os.path.exists(path): o..原创 2021-06-18 11:06:41 · 773 阅读 · 0 评论 -
BatchNorm1d
参考:https://zhuanlan.zhihu.com/p/100672008https://www.jianshu.com/p/2b94da24af3b# python3.8# -*- coding: utf-8 -*-# ---# @Software: PyCharm# @File: test2.py# @Author: ---# @Institution: BeiJing, China# @E-mail: lgdyangninghua@163.com# @Site:原创 2021-05-19 22:05:30 · 1274 阅读 · 0 评论 -
mmdetectionv2.8 踩坑
以下是自己配置v2.8遇到的一些问题mmdetv2.8今天首次尝试遇到的问题1.最新版v2.8使用mmcv-full,不再是mmcv2.mmcv-full现在修改为与CUDA和pytorch具备对应性,版本安错了也不行3.我的环境是cuda10.0 torch1.4.0 在其官方的下载地址上,没有该文件https://download.openmmlab.com/mmcv/dist/cu100/torch1.4.0/index.html里面没有mmdetv2.8用到的mmcv==1.2.4.原创 2021-05-07 15:19:05 · 2648 阅读 · 4 评论 -
mmdetection v2.8跑视频保存
转载:https://www.freesion.com/article/73351406450/转载 2021-05-07 12:23:15 · 256 阅读 · 0 评论 -
torch-points3d(1.2.0版本)示例代码
1分类代码示例RSConvCLassifier.py# python3.8# -*- coding: utf-8 -*-# ---# @Software: PyCharm# @File: RSConvCLassifier.py# @Author: ---# @Institution: BeiJing, China# @E-mail: lgdyangninghua@163.com# @Site: # @Time: 4月 27, 2021# ---import torchf.原创 2021-04-29 13:41:31 · 1615 阅读 · 0 评论 -
windows10+VS2019+detectron2
参考:win10安装detectron2https://blog.csdn.net/iamleifeng/article/details/103877581Windows10下GPU版本的pytorch1.6.0+torchvision0.7.0安装教程(支持CUDA10.2,含whl快速下载地址)https://blog.csdn.net/xylbill97/article/details/108357819Windows10下安装detectron2超详细教程(小白视角)http原创 2021-04-07 00:54:20 · 1015 阅读 · 0 评论 -
mmdetectionv2.8初体验
不能同时安装mmcv和mmcv-full1.最新版v2.8使用mmcv-full,不再是mmcv2.mmcv-full现在修改为与CUDA和pytorch具备对应性,版本安错了也不行3.我的环境是cuda10.0 torch1.4.0 在其官方的下载地址上,没有该文件https://download.openmmlab.com/mmcv/dist/cu100/torch1.4.0/index.html里面没有mmdetv2.8用到的mmcv==1.2.4强行安装会一直卡在 Buildin.原创 2021-02-04 16:52:42 · 945 阅读 · 1 评论 -
simple-faster-rcnn-pytorch
simple-faster-rcnn-pytorch从编程实现角度学习Faster R-CNN(附极简实现)https://zhuanlan.zhihu.com/p/32404424西红柿炒蛋就加糖!https://www.cnblogs.com/kerwins-AC/category/1301570.html三年一梦https://www.cnblogs.com/king-lps/category/1208643.html逐字理解目标检测simp...原创 2021-02-02 15:34:19 · 750 阅读 · 0 评论 -
mmdetection-coco-实现单个图像的Recall和Precision计算(Recall是想找出漏检了哪些badcase,Precision是误检了哪些)
目标检测的评价指标(TP、TN、FP、FN、Precision、Recall、IoU、mIoU、AP、mAP)https://www.cnblogs.com/geoffreygao/p/14095701.html目标检测的评价指标(TP、TN、FP、FN、Precision、Recall、IoU、mIoU、AP、mAP)https://blog.csdn.net/xinyanh/article/details/102733730mmdetection_inference.pytemp_b原创 2021-01-13 15:29:28 · 6479 阅读 · 9 评论 -
mmdetection-可视化按类别颜色显示结果(左边显示gt右边显示pred)-输出单个类别的APAR-通过计算单张图片的recall进行划分badcase
mmdetection_inference.py# -*- coding:utf-8 -*-import osimport argparseimport cv2import mmcvimport numpy as npfrom pycocotools.coco import COCO, maskUtilsfrom mmdet.apis import init_detector, inference_detectorfrom temp_coco_utils import coco_e..原创 2020-12-16 14:55:13 · 2982 阅读 · 2 评论 -
coco数据集通过不同文件名id挑选train和test----coco数据集将json文件按不同类别分割成类别json
1.coco数据集通过不同文件名id挑选train和test# -*- coding: utf-8 -*-import osimport cv2import jsonimport argparseimport numpy as npimport randomfrom pycocotools.coco import COCO, maskUtilsdef main(args): #测试集 L-id R-id 每个id两个数据 test_key = [str(m_key)原创 2020-12-09 17:56:40 · 1452 阅读 · 1 评论 -
mmdetection-inference-画出bbox和mask
可视化左边为原图,右边为结果图如果有GT-json,可画出jsonimport osimport cv2import mmcvimport torchimport numpy as npfrom pycocotools.coco import COCO, maskUtilsfrom mmdet.apis import init_detector, inference_detector, show_resultfrom mmdet.apis import init_dist, .原创 2020-12-08 16:57:28 · 2829 阅读 · 4 评论 -
计算数据集均值方差
参考:https://blog.csdn.net/weixin_41765699/article/details/100118660对于数据集获取简单进行修改import osimport cv2import numpy as npdef searchDirFile(rootDir, list_, path_, endswith_): for dir_or_file in os.listdir(rootDir): filePath = os.path.join.原创 2020-12-08 12:34:17 · 845 阅读 · 0 评论 -
pytorch项目Student-Teacher anomaly detection修改训练方式,能将大图划分成块载入
项目代码:https://github.com/denguir/student-teacher-anomaly-detection其实也可以直接随机crop大图区域,然后再crop-patch(65*65)但是这里我们将大图划分成4块区域去做了before_dir = os.path.abspath(os.path.join(os.getcwd(), ".."))print("before_dir", before_dir) dataset = AnomalyDataset(cs原创 2020-11-25 13:26:22 · 1072 阅读 · 0 评论 -
pytorch1.1.0和1.3.1的版本之间一个小差异(torch.mean(output, axis=1))
pytorch1.3.1:def compactness_loss(output): # dim: (batch, vector) _, n = output.size() avg = torch.mean(output, axis=1) std = torch.std(output, axis=1) zt = output.T - avg zt /= std corr = torch.matmul(zt.T, zt) / (n - 1)原创 2020-11-19 11:54:46 · 1173 阅读 · 0 评论 -
ubuntu训练习惯记录(abspath,os.getcwd(),nohup等)
#!/usr/bin/env bashdataset="$1"GPUNAME="$2"PYTHON=~/anaconda3_py3.7/bin/pythonnohup $PYTHON teacher_training.py dataset GPUNAME > result_001.log 2>&1 &import osimport sysDATASET = sys.argv[1]os.environ['CUDA_VISIBLE_DEVICES'] = s.原创 2020-11-18 20:52:38 · 308 阅读 · 0 评论 -
异常检测资料整理
几个异常检测的汇总github:https://github.com/hoya012/awesome-anomaly-detectionhttps://github.com/yzhao062/anomaly-detection-resources工业异常检测数据集: 4G+https://www.mvtec.com/company/research/datasets/mvtec-ad/文章:https://zhuanlan.zhihu.com/p/90637193ht..原创 2020-11-18 12:24:22 · 754 阅读 · 0 评论 -
py3-coco数据集-获取mask区域做简单的数据增强-可视化显示coco-coco转换labelme-labme转换coco-递归路径找文件-自己格式转labelme
以下代码是一张图中只有一个对象的处理,关于一张图有多个对象,方法类似,需要遍历原始anns,大家可根据需求更改即可# -*- coding: utf-8 -*-import osimport sysimport getoptimport jsonfrom pycocotools.coco import COCO, maskUtilsimport cv2import numpy as npimport mathimport numpy as npnp.random.seed(10.原创 2020-10-16 11:59:19 · 1266 阅读 · 1 评论 -
miniconda-python3.6(mmdetection-1.0rc0)
conda、miniconda、anaconda的区别以及在pycharm中选择conda的虚拟环境https://blog.csdn.net/qq_18668137/article/details/80807829https://docs.conda.io/en/latest/miniconda.html下载地址anaconda各版本下载地址https://repo.continuum.io/archive/miniconda各版本下载地址https://repo.cont..原创 2020-10-13 13:22:08 · 1107 阅读 · 0 评论 -
ImportError: libcusparse.so.10.0: cannot open shared object file: No such file or directory
一般出现这种情况,肯定是使用pycharm,然后他找不到cuda虽然你在~/.bashrc 中已经设置好了:export CUDA_HOME=/usr/local/cudaexport PATH=/usr/local/cuda/bin:$PATHexport LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH但是还是会出问题,因为pycharm并没有关联到网上有一些建议修改pycharm.sh我觉得不妥,万一想换cuda怎么办原创 2020-09-24 11:08:42 · 2299 阅读 · 0 评论 -
mmdetection-2.1.0训练数据
0.遇到训练问题RuntimeError: Expected to have finished reduction in the prior iteration before starting a new one. This error indicates that your module has parameters that were not used in producing loss. You can enable unused parameter detection by (1) passi原创 2020-07-03 15:30:04 · 1142 阅读 · 1 评论 -
计算神经网络学习图像的仿射变换
DeepImageHomographyEstimationhttps://blog.csdn.net/zcg1942/article/details/85254077https://blog.csdn.net/ajing2014/article/details/53998866https://arxiv.org/abs/1606.03798https://github.com/yishiliuhuasheng/deep_image_homography_estimationhttp...原创 2020-05-09 17:38:55 · 1502 阅读 · 0 评论 -
python 图像数据增强库
https://www.aiuai.cn/aifarm422.htmlhttps://github.com/albu/albumentationshttps://albumentations.readthedocs.io/en/latest/https://github.com/bethgelab/imagecorruptionshttps://imgaug.readth...原创 2020-04-29 13:17:23 · 922 阅读 · 0 评论 -
tensorflow中的datasets以及自定义的数据增强--1.14.0版本
之前看tensorflow的代码时候,遇到http://d0evi1.com/tensorflow/datasets/原创 2020-04-28 11:59:42 · 1438 阅读 · 0 评论 -
孪生网络--图像对--变化检测(参考资料)
CDnet 2014: An Expanded Change Detection Benchmark DataseFULLY CONVOLUTIONAL SIAMESE NETWORKS FOR CHANGE DETECTIONStreet-View Change Detection with Deconvolutional NetworksChangeNet: A Deep Learnin...原创 2020-04-24 15:19:49 · 1001 阅读 · 0 评论 -
tensorflow保存模型和模型部署
https://cv-tricks.com/tensorflow-tutorial/save-restore-tensorflow-models-quick-complete-tutorial/https://github.com/tensorflow/tensorflow/blob/r1.13/tensorflow/python/tools/freeze_graph.py咫片...转载 2020-04-23 10:28:29 · 496 阅读 · 0 评论 -
tensorflow单机多卡训练
之前没怎么解除tensorflowhttps://github.com/tensorflow/models/tree/r1.13.0/tutorials/image【分布式训练】单机多卡的正确打开方式(一):理论基础https://fyubang.com/2019/07/08/distributed-training/【分布式训练】单机多卡的正确打开方式(二):TensorFlow...原创 2020-04-20 16:18:50 · 1641 阅读 · 0 评论 -
人脸对齐算法PRNet--face3d库(c++扩展实现和仅仅numpy实现两个版本)
https://github.com/YadiraF/PRNet人脸对齐算法PRNethttps://www.jianshu.com/p/b460e99e03b0三维人脸重建PRnet,3DDFA,2DASL哪个效果更好?https://www.zhihu.com/question/351676110PRNet人脸重建学习笔记https://blog.csdn.net...转载 2020-04-16 10:44:55 · 801 阅读 · 0 评论 -
关于人脸关键点的数据集WFLW数据预处理
下载地址:https://wywu.github.io/projects/LAB/WFLW.html我刚开始不知道有官方的,自己写了一个,然后发现,有的landmark比框大,使用最下外接矩形改了一下,先上原始的:import osimport cv2import numpy as np# self.img = cv2.imread(self.line[0])# sel...原创 2020-04-15 19:48:28 · 9857 阅读 · 4 评论 -
请问注意力机制中生成的类似热力图或者柱状图是如何生成的?
https://www.zhihu.com/question/274926848?sort=created转载 2020-03-24 17:36:46 · 5493 阅读 · 1 评论