- 博客(12)
- 收藏
- 关注
原创 cuda(c,py,sk)
pycuda安装找不到cuda.hC_INCLUDE_PATH=/usr/local/cuda/includeLIBRARY_PATH=/usr/local/cuda/lib64LD_LIBRARY_PATH=/usr/local/cuda/lib64exportC_INCLUDE_PATH=/usr/local/cuda/include:$C_INCLUDE_PATHexpor...
2019-08-03 13:41:29 142
原创 onnx
from torch.autograd import Variableimport torch.onnximport torchvisiondummpy_input = Variable(torch.randn(1,3,224,224)).cuda()model = torchvision.models.vgg16(pretrained=True).cuda()torch.onnx.ex...
2019-06-27 19:15:49 814
原创 numpy笔记
np.histogram()对一维数组进行直方图统计,其参数列表如下:Histogram(a,bins=10,range=None,normed=False,weights=None)其中,a是保存待统计数据的数组,bins指定统计的区间个数,即对统计范围的等分数。 range是一个长度为2的元组,表示统计范围的最小值和最大值,默认值为None,表示范围由 数据的范围决定,即(a.min(...
2019-06-25 19:03:58 474
原创 AI知识基础笔记
在概率论或信息论中,KL散度( Kullback–Leibler divergence),又称相对熵(relative entropy),是描述两个概率分布P和Q差异的一种方法。它是非对称的,这意味着D(P||Q) ≠ D(Q||P)。特别的,在信息论中,D(P||Q)表示当用概率分布Q来拟合真实分布P时,产生的信息损耗,其中P表示真实分布,Q表示P的拟合分布。...
2019-06-25 19:02:00 661
原创 Pytorch-读数据 or 模型
val_tf = pretrainedmodels.utils.TransformImage( model, scale=scale, preserve_aspect_ratio=args.preserve_aspect_ratio)val_loader = torch.utils.data.DataLoader( datasets.ImageFolder(...
2019-06-25 19:00:58 245
原创 shell常用操作笔记
1.指定文件下文件数,不包含文件夹ls -l ${指定的目录} | grep "^-" | wc -l2.指定文件下文件数,包含文件夹ls -lR ${指定的目录} | grep "^-" | wc -l3.指定文件下文件夹(目录)的个数,不包括子目录ls -l ${指定的目录} | grep "^d" | wc -l4.指定文件下文件夹(目录)的个数,包括子目录ls ...
2019-06-25 18:58:50 83
原创 Caffe SourceCode Learning-solve()
初始化完Solver类之后,调用基类Solver::Solve()进行迭代template <typename Dtype>void Solver<Dtype>::Solve(const char* resume_file) { CHECK(Caffe::root_solver()); LOG(INFO) << "Solving " <&l...
2018-12-03 07:19:38 90
原创 Caffe SourceCode Learning-Sovler
在train中调用CreateSolver(solver_param))初始化一个具体的solvershared_ptr<caffe::Solver<float> > solver(caffe::SolverRegistry<float>::CreateSolver(solver_param)); (1)基础的Caffe框架目前支持如下的...
2018-12-02 17:29:38 168
原创 Caffe SourceCode Learning-main
Caffe的windows安装和Linux安装在这里就不介绍了,网上有很多很好的教程。 1.Caffe框架的入口为./tools/caffe.cpp的main()(1)框架使用gflag对命令行参数进行解析DEFINE_type(p1,p2,p3)定义了需要从命令行传递参数的变量:p1为变量,p2默认值,p3为解释。GlobalInit()函数中调用了ParseCommandLi...
2018-12-02 10:01:12 116
原创 目标检测-Review
目标检测 = 目标定位 + 目标分类:利用图像处理技术、 机器学习等多方向的知识, 从图像(视频)中定位感兴趣的对象。目标分类负责判断输入的图像中是否包含所需物体(object), 目标定位则负责表示目标物体的位置, 并用外接矩形框定位。不管是哪种方法,想要从一个集合中找到一个物体,就是要抓住它区别于其他物体的地方:特征。传统目标检测算法的典型代表:1.SIFT算法(Scale-inv...
2018-11-21 06:55:07 296
原创 faster-rcnn学习
py-faster-rcnn地址:https://github.com/rbgirshick/py-faster-rcnnNetscope地址: http://ethereon.github.io/netscope/#/editorfaster rcnn有两种训练方式:1.交替优化方法(alternating optimization),分别在两个stage即训练rpn网络和fast ...
2018-11-19 22:55:49 136
原创 faster-rcnn学习--stage1_rpn_train.pt
第一阶段的rpn训练是在pre-trained模型的基础上fine-tuning print '~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~' print 'Stage 1 RPN, init from ImageNet model' print '~~~~~~~~~~~~~~~~~~~~~~~~~...
2018-11-18 22:53:16 439
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人