![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
深度学习
wi162yyxq
这个作者很懒,什么都没留下…
展开
-
YOLOV5 onnxruntime预测不用pytorch
使用onnxruntime不使用pytorch完成yolov5推理原创 2021-11-23 13:42:10 · 2893 阅读 · 3 评论 -
win10安装wsl2及nvidia-docker遇到的一些个问题
最近双系统崩了,所以选择wsl2得了,省得以后麻烦,首先是安装方法,这里贴上我参考的一些帖子及简介:win10安装wsl2及docker-ce的方法:https://blog.csdn.net/xianxi9883/article/details/107358445/win10开启容器虚拟化:https://blog.csdn.net/leenhem/article/details/105359112http://www.xitongcheng.com/jiaocheng/win10_a原创 2020-10-02 10:26:19 · 4913 阅读 · 4 评论 -
U2NET人物肖像画模拟
论文:U^2-Net: Going Deeper with Nested U-Structure for Salient Object Detection - 2020作者:Xuebin Qin, Zichen Zhang, Chenyang Huang, Masood Dehghan, Osmar R.Zaiane, MartinJagersand团队:University of Alberta, Edmonton, CanadaGithub -U-2-NetU2NET其实就是将一个un.原创 2020-09-06 09:46:43 · 980 阅读 · 2 评论 -
xml转coco格式
使用方法极其简单,将xml文件和原始图像文件放到同一个文件夹,然后在代码里改一下xml地址就可以了。#coding:utf-8 # pip install lxml import osimport globimport jsonimport shutilimport numpy as npimport xml.etree.ElementTree as ETimport cv2 path2 = "." START_BOUNDING_BOX_ID = 1 def原创 2020-07-05 19:19:15 · 863 阅读 · 0 评论 -
xml转coco格式代码
只需要将labelme等标注工具标注的xml和原始图像文件放在一个文件夹里,把代码里的文件夹位置改了就可以了,很简单,转换完了就可以直接用来训练了。#coding:utf-8 # pip install lxml import osimport globimport jsonimport shutilimport numpy as npimport xml.etree.ElementTree as ETimport cv2 path2 = "." START_BOU原创 2020-06-30 15:12:00 · 3440 阅读 · 7 评论 -
DBface
网络其实还是蛮简单的,多尺度进行人脸检索,最厉害的地方就是模型只有7MB,速度很快,我的破电脑只有不到2G显存都能实时检测。使用方法也极其简单def image_demo(): dbface = DBFace() dbface.eval() if HAS_CUDA: dbface.cuda() dbface.load("model/dbfaceSmallH.pth") detect_image(dbface, "datas/se...原创 2020-05-26 13:50:05 · 878 阅读 · 0 评论 -
pytorch梯度累加反向传播
传统的训练函数,一个batch是这么训练的:for i,(images,target) in enumerate(train_loader): # 1. input output images = images.cuda(non_blocking=True) target = torch.from_numpy(np.array(target)).float().cuda(non_blocking=True) outputs = model(images) los转载 2020-05-11 15:05:05 · 1045 阅读 · 0 评论 -
libtorch调用模型封装so,并使用Python调用
C++代码如下#include "torch/script.h"#include "torch/torch.h"#include <opencv2/opencv.hpp>#include <iostream>#include <memory>#include <string>#include<cstdio>#incl...原创 2020-03-22 10:12:12 · 1803 阅读 · 3 评论 -
pytorch中attention的两种实现方式
class AttnDecoderRNN(nn.Module): def __init__(self, hidden_size, output_size, dropout_p=0.1, max_length=MAX_LENGTH): super(AttnDecoderRNN, self).__init__() self.hidden_size = hi...原创 2019-12-18 16:32:51 · 20931 阅读 · 0 评论 -
CTC反向传播
记录一下CTC反向传播,在此推荐谷歌的一个CTC讲解的PPT,讲的很清楚https://docs.google.com/presentation/d/12gYcPft9_4cxk2AD6Z6ZlJNa3wvZCW1ms31nhq51vMk/pub?start=false&loop=false&delayms=3000&slide=id.g24e9f0de4f_0_201...原创 2019-08-07 10:18:39 · 331 阅读 · 0 评论 -
C#调用tensorflow模型EAST识别文字
主要利用了Emgucv这个opencv的C#库,但是由于Emgucv里的NMS只能计算Rect不能计算RotateRect所以识别水平文字还行,倾斜的就不行了,代码如下:(写的不规范什么的勿喷....如果有什么问题请告诉我,互相提高嘛)using System;using System.Collections.Generic;using System.Linq;using System...原创 2019-07-16 17:43:30 · 2134 阅读 · 1 评论 -
创建tf_record文件(直观版)
import os import tensorflow as tf from PIL import Image #注意Image,后面会用到import matplotlib.pyplot as plt import numpy as npwriter= tf.python_io.TFRecordWriter("cifar10_train.tfrecords")writer_2...原创 2019-03-04 18:55:13 · 421 阅读 · 0 评论 -
module 'setuptools.dist' has no attribute 'check_specifier'
安装maskrcnn遇到如题问题,查询后得到的解决方法是:pip install --upgrade piphash -r pippip3 install --upgrade piphash -r pip3cd (Mask RCNN root folder)sudo python3 setup.py install成功解决,撒花~~~...原创 2019-02-25 20:39:35 · 763 阅读 · 0 评论