- 博客(18)
- 收藏
- 关注
原创 torch.LongTensor转换成one hot tensor编码
import torch'''pytorch0.4.0及以上的高版本,加上pytorch0.2.0都支持torch.Tensor._scatter功能实现将torch.LongTensor编码成 one-hot vector的功能'''rois_label=torch.tensor( [[0,1,0,5,2,7,4,3,1,2]])#假设当前的ground truth box...
2019-02-27 10:24:58 5102
原创 pytorch 图像预处理之减去均值,除以方差
#coding=gbk'''GPU上面的环境变化太复杂,这里我直接给出在笔记本CPU上面的运行时间结果由于方式3需要将tensor转换到GPU上面,这一过程很消耗时间,大概需要十秒,故而果断抛弃这样的做法img (168, 300, 3)sub div in numpy,time 0.0110sub div in torch.tensor,time 0.0070su...
2019-02-25 21:06:46 9020 2
原创 focal loss和OHEM(on-line hard example mining)如何应用到faster RCNN中
在物体检测问题中,主要分为两类检测器模型:one stage detector(SSD,YOLO系列,retinanet)和two stage detector(faster RCNN系列及其改进模型),然而无论是一个阶段的检测器还是两个阶段的检测器,都使用到了anchor机制,即在特征图上密集地画anchor boxes,根据先验知识设定的IOU阈值将这些anchor划分为正样本和负样本,再对于...
2019-02-25 17:47:27 5591 24
原创 SSD与Faster RCNN中的RPN生成anchor的不同之处
SSD是在多个不同尺度/分辨率的特征图上生成anchor,故而在每一个尺度的特征图上生成anchor时,并没有引入multi-scale,而是给同一个尺度分辨率特征图上的每个像素点设定一个面积值,在该像素点上产生 len(aspect_ratios)个anchor boxes,即产生不同宽高比的anchor boxes,这些不同宽高比的anchor boxes在同一个分辨率的特征图上具有相同的面积...
2019-02-23 22:04:19 2116
原创 PIL.Image与cv2读取图像时间的测试
from PIL import Imageimport cv2import timeimg_path='F:\\2\\scene.jpg'cv_start=time.time()#2.使用cv2读取彩色图像img_CV=cv2.imread(img_path)# print(img_CV.shape,type(img_CV))#(168, 300, 3) <class 'n...
2019-02-22 15:39:56 1398
原创 YOLO V3 (一)
https://blog.paperspace.com/how-to-implement-a-yolo-object-detector-in-pytorch/首先明确,YOLO 是一个全卷积网络(fully convolutional neural network,FCN),其中的下采样操作并不使用pooling层,而是使用步长stride=2的卷积操作代替,以避免对于低阶高分辨率特征图的信息...
2019-02-20 12:35:29 972
转载 bounding box regression in detection model
http://caffecn.cn/?/question/160https://blog.csdn.net/zijin0802034/article/details/77685438/
2019-02-18 21:14:26 143
原创 python实现对于数据集的划分(随机划分出训练集和验证集)
import osimport random'''此程序是将所有的带有标签的数据集划分为训练数据集和验证数据集按照一定比例从整个数据集中随机挑选出验证数据集,剩下的部分作为训练数据集从裁剪之后的所有图像集合中随机采样出1/9的图像作为测试图像,剩下的8/9 作为训练图像其中训练图像:测试图像 比例与之前相同'''sub_path='F:\\g\\detection_sub'f...
2019-02-17 20:19:29 15377
原创 numpy ndarray 取出满足特定条件的某些行
在进行物体检测的ground truth boxes annotations包围框坐标数据整理时,需要实现这样的功能:numpy里面,对于N*4的数组,要实现对于每一行,如果第3列和第1列数值相等或者第2列和第0列数值相等,就删除这一行,要返回保留下来的numpy数组 shape M*4对于numpy数组的操作要尽量避免for循环,因为numpy数组支持布尔索引。import num...
2019-02-17 15:51:14 8931
原创 numpy中的行向量和列向量
import numpy as npa=np.array([1,2,3])#1.产生行向量的方法row_vector1=a.reshape(1,-1)row_vector2=np.array([[1,2,3]])row_vector3=np.expand_dims(a,0)print(row_vector1.shape,row_vector2.shape,row_vector3.sh...
2019-02-17 15:31:38 4159
原创 pycharm打不开
今天pycharm突然就打不开了,卸载后重新安装仍然打不开。解决办法如下:打开命令提示符,输入 netsh winsock reset 这个命令,然后重启电脑,就解决了
2019-02-16 17:03:11 13909 10
原创 GPU nvidia-smi
服务器上有多块GPU,每块GPU上可能运行着多个python程序,之前在调试程序时,每次想要kill掉正在运行的程序,我都是 ps -ef | grep xxx.py 这个命令会从当前所有正在运行的python进程中找到名称为xxx.py的进程,给出所有父进程ID和子进程ID,通常做法是先kill掉父进程PID,再一个个地kill掉子进程PID,这样的做法看似有些复杂。‘今天问过一...
2019-02-15 22:40:52 584
原创 python write和writelines
import numpy as npa=np.random.randn(2,3)# print(a)'''numpy.random模块产生随机numpy.ndarraynumpy.random.randn 产生的数组中数值服从正态分布numpy.random.rand 产生的数组中数值服从0-1之间的均匀分布'''out_txt='F:\\2\\out.txt'w...
2019-02-15 21:58:14 2201
原创 torch.max
import torcha=torch.randn(5,3)print(a,torch.max(a,1),torch.max(a,1)[1])'''tensor([[ 1.0588, 0.4321, -1.1318], [-1.5585, -0.5181, -0.6880], [ 2.1610, 1.1872, -0.7473], [-...
2019-02-15 20:33:04 150
原创 SyntaxError: Non-UTF-8 code starting with '\xca'
解决方法,在 ***.py文件首行添加代码# coding=gbk
2019-02-11 15:22:31 2046 1
原创 python中将数值写入字符串的两种方式
a=int(5)b=float(4.3)c=0.005str1='%d %.4f %.2e'%(a,b,c)#这种方式可以在一个长字符串中同时插入多个数值型数据str2='int is {}'.format(str(a))#使用format方式只允许插入一个,用字符串代替{}花括号print(str1)print(str2)'''5 4.3000 5.00e-03int ...
2019-02-11 15:10:59 3289
原创 pytorch CUDA out of memory
之前用FPNSSD训练时遇到CUDA out of memory的报错,通常通过减小batch size,或减小输入图像的空间分辨率来解决,这此次我训练faster R-CNN时,设置了batch size=1 ,保证输入图像的最大空间分辨率不超过1280,但是开始几个epoch训练正常(已经出现过经过变换后允许的最大空间分辨率——1280*1280),但是再训练几个epoch,就会随机不定时地产...
2019-02-11 14:05:00 1974
原创 python3 libtiff安装
总结这种技术上面的经验,我需要感谢身边的大神,他教会我,研究生的阶段其实就是:发现问题,分析问题和解决问题的过程,老师常跟我说,它是不是一个问题由自己决定,我常常总认为工程技术上的问题它不是问题,想换更多精力关注于算法本身,跟着老师半年时间让我对问题的定义有了很大改变,工程实现上的问题往往更不能被忽略。因为训练一个数据集进行segmentation task training的过程中,首先要对...
2019-02-04 15:04:20 5110 4
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人