- 博客(32)
- 收藏
- 关注
RSS订阅原创 torch版本不一致，网络前向导致网络输出维度不一致问题
在torch1.7下，偶然发现使用ResNet框架，在前向的时候，最后一层总是未被使用。使用相同的测试代码（有兴趣可以自己复制测试代码到相应环境下测试）：import torchimport torch.nn as nnimport torchvisioninputVar = torch.randn(1, 3, 224, 224)model = torchvision.models.resnet50(pretrained=True)num_ftrs = model.fc.in_features
2021-10-26 11:14:59
1008
原创 python中json报错JSONDecodeError: invalid control character
解决方案：json.loads(json, strict=False)#关掉strict即可原因：json.loads 解析时会检查返回的数据是否完全满足json格式，如有特殊意义字符会导致json字符串格式变化，不满足完全的json格式，所以会报错...
2020-12-06 15:43:20
1340
1
原创 2020-11-24
typedef Vec<uchar, 3> Vec3bint color_dis(Vec3b color1, Vec3b color2){ int rmean = (color1[2] + color2[2]) / 2 int R = color1[2] - color2[2] int G = color1[1] - color2[1] int B = color1[0] - color2[0] int distance = (2 + rmean /
2020-11-24 21:51:22
234
原创 统计学三大相关系数---Pearson、Spearman、kendall / RMSE与实现（附代码）
算法实现：这里直接调库from scipy import statsimport numpy as npval_PLCC = stats.pearsonr(y_pred, y_val)[0]val_SROCC = stats.spearmanr(y_pred, y_val)[0]val_RMSE = np.sqrt(((y_pred-y_val) ** 2).mean())val_KROCC = stats.stats.kendalltau(y_pred, y_val)[0]一、PLCC（P
2020-11-06 17:22:21
6311
原创 黑边检测|蒙版检测 边缘检测 霍夫曼直线检测（附代码）
一、背景介绍实际应用场景中，图片或视频中可能会存在黑边，影响美观度，不适合推荐给用户，本文主要是介绍如何去除黑边和蒙版。（限于篇幅，以图像蒙版去除为例子，进行讲解）(1) 先看下效果A图和C图是带有蒙版的的视频帧（图像），B图和D图是检测蒙版并进行裁剪后的图像，可以看出，该算法可以完美去除蒙版。【黑边去除更易】下面的分数表示基于蒙版的遮盖程度，对该图片进行打分，分数越高，说明视频被遮挡的区域越小（2）算法流程（a）先用candy算子进行边缘检测（b）使用霍夫曼直线检测，检测直线（c）NM
2020-10-30 11:41:02
4073
2
原创 C++读取gif图 | opencv读取gif图(附完整代码)
一、背景介绍：opencv无法对gif进行直接读取，在python中PIL可以读取，但是在C++中，目前不知道如何使用PIL库，另外有推荐说使用FreeImage，但是文档不全使用非常麻烦，故使用以下方法解码gif的每一帧，并存放到frames里面。简单、方便、有效！二、代码:main.cpp#include<iostream>#include<opencv2/opencv.hpp>#include<vector>using namespace cv;us
2020-10-30 11:12:57
7513
4
原创 C++检查内存泄漏问题
参考：https://zhuanlan.zhihu.com/p/75416381valgrindvalgrind --tool=memcheck --leak-check=fulldefinitely lost: 1 bytes in 1 blocks
2020-09-17 20:22:00
155
原创 C++线程锁
参考资料：1、在CmakeList里面加入：## openmp opencvfind_package(OpenMP)if(OpenMP_CXX_FOUND OR OPENMP_FOUND) message(STATUS "===========OpenMP module============") set(CMAKE_C_FLAGS "${CMAKE_C_FLAGS} ${OpenMP_C_FLAGS}") set(CMAKE_CXX_FLAGS "${CMAKE_CXX_F
2020-09-17 20:18:43
266
原创 pytorch的18个损失函数
PyTorch 学习笔记（六）：PyTorch的十八个损失函数 - 余霆嵩的文章 - 知乎https://zhuanlan.zhihu.com/p/61379965
2020-06-23 14:39:15
365
原创 面试编程
面试知识点总结https://blog.csdn.net/oBrightLamp/article/details/850679811. 实现NMS目标： 加快检测速度，使用NMS去除冗余框，找到具有代表性的bbox，进行后续操作算法步骤：根据候选框的类别分类概率做排序：A<B<C<D<E<Fstep1：先标记最大概率矩形框F是我们要保留下来的；ste...
2020-05-16 19:10:04
222
原创 统计视频时长VideoFlipClip--在VideoFileClip函數中獲取“OSError：[WinError 6]句柄無效”
import osimport datetimeimport sysimport argparsefrom moviepy.editor import VideoFileClipif __name__ == "__main__": parser = argparse.ArgumentParser( description='Compute Total Time ...
2020-02-25 09:22:29
504
原创 docker使用
公司docker使用简单介绍（入门）1. docker镜像1.方法1： 编写docker_run.sh 拉取别人的镜像编写docker_run.sh，执行 sh docker_run.sh 即可进入容器。docker_run.sh内容如下： 1 set -x 2 set -e 3 4 ROOT_DIR=$(cd "$(dirname "$0")";pwd) #换成你需要...
2019-10-30 21:33:52
461
原创 多任务学习 相关方法
1. 定义：基于共享表示（shared representation），把多个相关的任务放在一起学习的一种机器学习方法。2. 多任务学习好处：i. 提高泛化能力多人相关任务放在一起学习，有相关的部分，但也有不相关的部分。当学习一个任务（Main task）时，与该任务不相关的部分，在学习过程中相当于是噪声，因此，引入噪声可以提高学习的泛化（generalization）效果。ii. 降低...
2019-08-13 20:49:55
1271
原创 linux建立软连接
创建软连接：ln -s a b 中的 a 就是源文件，b是链接文件名,其作用是当进入b目录，实际上是链接进入了a目录ps：执行命令的时候,应该是a目录已经建立，目录b没有建立。删除软链接：rm -rf b 注意不是rm -rf b/例子：在~/workspace/fgimg/NTS-Net目录下，链接/home/liyating/Mytask/datasets/CUB_...
2019-06-11 11:33:18
677
原创 python学习 知识点记录
知识点概要：函数式编程（高阶函数、闭包、匿名函数、装饰器)模块和包（模块、包）面向对象编程（类、实例、属性、方法)类的继承（目的、多态、多重继承)定制类（目的、方法、call）1.课程介绍2.函数式编程 (1) map函数 map(f,L)： 通过把函数 f 依次作用在L（ L是list） 的每个元素上，得到一个新的 list 并返回；例子：def f(x): ...
2019-06-10 11:21:22
194
原创 多进程与多线程
多线程：同时执行多项任务同时听音乐和看电影多进程：同一个任务由多个子机器同时完成，每个子机器均处理该任务的一部分1.多进程method1：手动创建step1：import multiprocessing 导入multiprocessing包step2：对象名 = Process(target=函数名) 创建Process类的对象，创建进程step3：对象名.start() 启动进程 ...
2019-06-09 17:24:36
140
原创 non_local论文阅读&代码
1.论文阅读 Non-local Neural Networks1.问题来源CNN中的 convolution单元每次只关注邻域 kernel size 的区域，就算后期感受野越来越大，终究还是局部区域的运算，这样就忽略了全局其他片区（比如很远的像素）对当前区域的贡献。2.主要思想non-local blocks 要做的是，捕获这种 long-range 关系：对于2D图像，就是图像...
2019-05-07 16:43:50
1882
原创 Videos as Space-Time Region Graphs阅读笔记
1.解决问题：基于I3D模型，使用图模型，以构造视频中物体间的关系，提升识别精度。本文是将video看做是 a graph of objects,然后再该graph上进行行为识别的推理。2.实验效果：在somethingV1数据集上，test：45% 相对于I3D 提高1.7% ，相对于TRN网络提高12%3.图卷积层 Convoltions on Graphs,graph conv ...
2019-04-13 22:10:52
913
3
原创 git使用步骤（附录：创建git分支）
emm~~一直没有管理代码的不良习惯，自从服务器突然崩了，代码都没了，我觉得是时候好好管理一下了！！！进入正题：注册github账号 邮箱 （3501477328@qq.com）+ 密码 （***）获取SSH Key的秘钥对本地执行：ssh-keygen -t rsa -C "youremail@example.com"会得到：id_rsa是私钥，不能泄露出去，id_rsa.pu...
2019-04-12 10:31:27
280
原创 安装caffe
caffe详细的安装步骤：安利博客：https://blog.csdn.net/qq_33039859/article/details/80377356检测caffe是否安装成功的小例子：博客 https://blog.csdn.net/weixin_42268842/article/details/81021614...
2019-04-04 18:51:52
124
原创 pytorch模型保存与加载以及常见问题
一. 模型保存与加载#多gpuimport osos.environ['CUDA_VISIBLE_DEVICES'] = '1,2,3,4' #choosemodel = TheModelClass(*args, **kwargs)model = torch.nn.DataParallel(model).cuda()#加载与训练模型 file.pth.tarcheckpoint =...
2019-04-04 12:15:48
7216
4
原创 pytorch中的Datasets&DataLoader&datasets.ImageFolder相关问题
一.自定义Datasetsclass CustomDataset(data.Dataset):#需要继承data.Dataset def __init__(self): # TODO # 1. Initialize file path or list of file names. def __getitem__(self, index): ...
2019-04-01 18:50:26
2695
原创 caffe与tensorflow之间模型转换
一.caffe转换为tensorflow1.下载模型转换代码:git clone https://github.com/ethereon/caffe-tensorflowcaffe-tensorflow模型转换文件为convert.py,可以将caffe的模型定义文件.prototxt和训练好的模型分别转换为tensorflow类,和tensorflow模型文件.2.举个例子将exam...
2019-03-31 22:06:33
2927
1
原创 tensorflow和pytorch模型之间转换
参考链接：https://github.com/bermanmaxim/jaccardSegment/blob/master/ckpt_to_dd.py一. tensorflow模型转pytorch模型import tensorflow as tfimport deepdish as ddimport argparseimport osimport numpy as npdef ...
2019-03-31 18:03:41
33288
38
原创 pytorch不同版本安装以及版本查看
一：基于conda安装conda create --name pytorch_learn python=3.6.7#创建一个名为pytorch_learn的环境source activate pytorch_learn #进入环境conda install pytorch=0.3.1 cuda80 -c soumith #安装pytorch0.3.1+ cuda8.0（可自己指定）cond...
2019-03-30 23:02:44
13789
5
原创 安装tensorflow-gpu详细步骤
cuda+cudnn+tensorflow-gpu一.安装cuda去官网https://developer.nvidia.com/cuda-toolkit-archive 选择要下载的cuda版本，如下：2.按照如下方式选择环境(这个好像不需要登录也可以下载)，选择base installer所在栏点击下载，得到cuda_9.0.176_384.81_linux.run文件。3.安装...
2019-03-29 09:03:07
1011
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人