2018魔方-CSDN博客

原创 Saliency map （fixation map）生成热点图-python实现

废话不多说，直接上代码：import cv2import numpy as npdef heat_map(path): fix = cv2.imread(path) max_val = np.max(fix) min_val = np.min(fix) fix = (fix - min_val)/(max_val - min_val) fix *= 255 pred_ = fix.astype(np.uint8) pred_heat_ma.

2020-09-19 12:05:50 2979 2

原创 Ubuntu下C++调用pytorch训练好模型--利用libtorch

最近因项目需要，需C++调用pytorch模型，以下是一些学习心得，把过程记录下来。搭建有以下几个流程：官网下载库libtorch pytorch模型转化编写C++调用程序一、官网下载库libtorch下载libtorch地址(https://pytorch.org/get-started/locally/），选择对应版本（根据自身电脑配置），比如我是Ubuntu+CUDA 10.2下载好之后，选择路径进行解压（注意自己下载的包），命令如下：unzi

2020-07-22 11:12:30 3448 2

原创语义分割标注文件json转mask图片示例

注：本文主要参考博客http://neroasmar.top/labelme-batch/，感谢作者的分享。我们平时用labelme标注图片得到的都是json文件，而实验需要用到的是Mask图；labelme自带的命令labelme_json_to_datasetlabelme_json_to_dataset xxx.json每次只能转化单个文件，如果要转化多个文件比较麻烦，并且不...

2020-04-29 10:25:04 5991 1

原创学习笔记--机器学习之牛顿法和拟牛顿法

牛顿法和拟牛顿法都是解决无约束问题最优化的方法，也是比较常见的方法之一。我们知道随机梯度下降，采用目标函数一阶导更新参数，但存在的问题很多，比如收敛速度慢、收敛到局部最优而不是全局最优或者无法收敛等问题。而本文提到的牛顿法不仅考虑了一阶导，还考虑目标函数二阶导来解决优化问题，收敛速度快，适用于小样本特征数据，因为在迭代每次时，需要求海塞矩阵的逆，当数据特征维度比较高时，需要的存储空间和计算成本都很...

2020-03-26 15:45:45 404

原创学习笔记--机器学习之提升方法（AdaBoost）

提升方法是一个线性组合多个不同的弱分类器使其成为强分类器的方法（结合多个分类器的优势），训练过程主要通过不断加大误分类样本权重，减小正确样本数据权重（改变训练数据概率分布）来达到最终的分类器。其中最为典型的是AdaBoost算法。补充概念：强学习：通过一个多项式学习（分类器）就可以达到比较高的正确率弱学习：通过一个多项式学习（分类器）仅能达到随机猜测相近的正确率AdaBoost需解决...

2020-03-24 11:09:21 267

原创计算两矩形的交并比IOU

# 计算两矩形的交并比IOU#[x1,y1,x2,y2],其中(x1,y1)为矩形左上角;(x2,y2)为矩形右下角def get_IOU(R1,R2): x1,y1,x2,y2=R1 m1,n1,m2,n2=R2 W=min(x2,m2)-max(x1,m1) H=min(y2,n2)-max(y1,n1) if W<=0 or H<...

2020-01-11 16:44:51 440

原创 Python 同一窗口显示多张图片（同时读取文件夹下所有图片）

因为百度了好几次才找到自己想要的东西，因此记录下来。一方面方便自己以后用；另一方面分享给大家。长话短说，直接贴代码：import cv2 as cvimport osimport numpy as npimg_path='./Track1_Saliency/Images/' #图像路径re_path='./Track1_Saliency/Results/' #图像路径img...

2019-06-20 09:40:22 13092

原创 tensorflow维度相关的问题

1、降掉为1的维度#P => shape=[1,3,3,1]P1=tf.squeeze(P) #降掉所有为1的维度P2=tf.squeeze(P, [3]) #降掉指定维度为1的维度#降维之后的结果P1 => Tensor("Squeeze:0", shape=(3, 3), dtype=float32)P2 => Tensor("Squeeze_1:0", ...

2019-06-05 09:35:08 254

转载转---图像显著性检测论文及代码汇总

转实验室师兄的，原文链接https://blog.csdn.net/u010736662/article/details/88930849PublicationPaperURL-Saliency Evaluation Metrics (AUC, sAUC, CC, NSS and KLD)code-MIT saliency benchmarkcode...

2019-03-31 20:35:45 2274

原创 Python文件名排序或文件排序问题

在读取文件夹里的文件时，有时需要顺序读取文件，那python如何顺序读取文件呢（这些都是楼主实际编程过程中遇到的问题，想把细节记录下来），下面会给出一些参考方式；特此说明：在编辑本文的时候，楼主也参考了一些文献，如有侵权，联系楼主，本人立马删帖。在python环境中提供两种排序方案（题主知道的）：用库函数sorted()对字符串排序，它的对象是字符用函数sort(...

2018-10-20 16:42:13 48289 4

原创 Win7安装CPU版TensorFlow各种神坑（已解决）

在安装tensorflow时，我们会遇到各种问题，比如python版本和tensorflow版本不兼容等问题；总之楼主经历各种坑才成功安装win下cpu版本的tensorflow，以下是我的一些经验总结，希望能帮助遇到同样问题的小白们1、首先安装python3.5.2（为什么安装这个版本，试了3.6版本都没有解决问题，并且3.6版本对应的TensorFlow好像不支持CPU版本的安装，这...

2018-09-14 16:51:01 4392

qq_36481821的博客