HOG:概述
HOG是什么?
方向梯度直方图,也称为HOG,是一种特征描述符,类似于Canny边缘检测器。它用于计算机视觉和图像处理中的目标检测。
该技术统计图像局部区域中梯度方向的出现次数。该方法类似于边缘方向直方图和尺度不变特征变换(SIFT)。
HOG描述符关注对象的结构或形状。它比任何边缘描述符都好,因为它使用梯度的大小和角度来计算特征。对于图像区域,它使用梯度的大小和方向生成直方图。
论文:Histograms of Oriented Gradients for Human Detection
计算特征的步骤
1.获取要计算其特征的输入图像。将图像调整为128x64像素(高128像素,宽64像素)的图像。论文中使用了该维度,作者建议将其作为此类检测的主要目标,以在行人检测任务中获得更好的结果。由于本文作者在麻省理工学院行人数据库上获得了异常完美的结果,他们决定制作一个新的、更具挑战性的数据集,称为“INRIA”数据集(http://pascal.inrialpes.fr/data/human/),包含从一组不同的个人照片中剪下的1805(128x64)幅人类图像。
图1:导入的图像。图2:灰度图像。图3:缩放导入图像和灰度图像:
2.计算图像的梯度。梯度是通过结合图像的大小和角度来获得的。考虑3x3像素的块,首先计算每个像素的Gx和Gy。对于每个像素值,首先使用以下公式计算Gx和Gy。
在计算Gx之后,使用下面提到的公式计算每个像素的幅值和角度。
图4:图像幅值的可视化。图5:图像角度的可视化图片
3.在获得每个像素的梯度后,梯度矩阵(幅值和角度矩阵)被划分为8x8个单元以形成块。对于每个块,计算9 区间 直方图,每个区间的角度范围为20度。图8表示一个9区间的直方图,其中的值在计算后分配。这些直方图中的各个区间输出该块中梯度的幅值。由于块包含64个不同的值,因此对于所有64个幅值和梯度值,将执行以下计算。由于我们使用9区间直方图,因此:
第j个区间的边界如下:
每个区间的中心值为:
图6:幅值图像上的8x8块。图7:角度图像上的8x8块图片
4.对于块中的每个单元,我们将首先计算第j个区间,然后计算将分别提供给第j个和(j+1)个区间的值。该值由以下公式给出:
5.将数组列作为区间的单元,并将Vj和Vj+1的值附加到数组中,索引为每个像素计算的第j个和第(j+1)个单元。
6.上述计算后的合成矩阵形状为16x8x9。
7.完成所有区间的直方图计算后,将9区间直方图矩阵中的4个区间组合在一起,形成一个新区间(2x2)。以重叠的方式移动,步幅为8像素。对于区间中的所有4个单元,我们将每个组成单元的所有9区间直方图连接起来,形成36个特征向量。
上图说明了9区间直方图的计算方法
在图像周围遍历2x2网格框,以便从4个bin生成组合fbi
8.每个bin的fbi值通过L2范数标准化:
式中,ε是一个小值,加在fb的平方上,以避免零除法误差。在代码中,取的值为1e-05。
9.为了规范化,首先通过以下公式计算k值:
10.进行此标准化是为了减少相同对象的图像之间对比度变化的影响。从每个block中,采集了一个36点的特征向量。在水平方向上有7个区间,在垂直方向上有15个区间。因此,HOG特征的总长度为:7 x 15 x 36=3780。获得所选图像的HOG特征。
使用skimage库对同一图像上的HOG特征进行可视化
Python计算HOG特征
导入库和所需的映像
import matplotlib.pyplot as plt
from skimage import io
from skimage import color
from skimage.transform import resize
import math
from skimage.feature import hog
import numpy as np
img = resize(color.rgb2gray(io.imread("B.jpg")), (128, 64))
所使用图像的可视化
plt.figure(figsize=(15, 8))
plt.imshow(img, cmap="gray")
plt.axis("off")
plt.show()
img = np.array(img)
计算图像的梯度和角度
mag = []
theta = []
for i in range(128):
magnitudeArray = []
angleArray = []
for j in range(64):
# 轴0的条件
if j-1 <= 0 or j+1 >= 64:
if j-1 <= 0:
# 第一个元素
Gx = img[i][j+1] - 0
elif j + 1 >= len(img[0]):
Gx = 0 - img[i][j-1]
# 第一个元素
else:
Gx = img[i][j+1] - img[i][j-1]
# 轴1的条件
if i-1 <= 0 or i+1 >= 128:
if i-1 <= 0:
Gy = 0 - img[i+1][j]
elif i +1 >= 128:
Gy = img[i-1][j] - 0
else:
Gy = img[i-1][j] - img[i+1][j]
# 计算幅度
magnitude = math.sqrt(pow(Gx, 2) + pow(Gy, 2))
magnitudeArray.append(round(magnitude, 9))
# 计算角度
if Gx == 0:
angle = math.degrees(0.0)
else:
angle = math.degrees(abs(math.atan(Gy / Gx)))
angleArray.append(round(angle, 9))
mag.append(magnitudeArray)
theta.append(angleArray)
mag = np.array(mag)
theta = np.array(theta)
图像幅度的可视化
plt.figure(figsize=(15, 8))
plt.imshow(mag, cmap="gray")
plt.axis("off")
plt.show()
图像角度的可视化
plt.figure(figsize=(15, 8))
plt.imshow(theta, cmap="gray")
plt.axis("off")
plt.show()
number_of_bins = 9
step_size = 180 / number_of_bins
计算第j个区间的函数
def calculate_j(angle):
temp = (angle / step_size) - 0.5
j = math.floor(temp)
return j
计算第j个区间的中心值的函数
def calculate_Cj(j):
Cj = step_size * (j + 0.5)
return round(Cj, 9)
计算第j个区间的值的函数
def calculate_value_j(magnitude, angle, j):
Cj = calculate_Cj(j+1)
Vj = magnitude * ((Cj - angle) / step_size)
return round(Vj, 9)
为8x8单元格提供9区间直方图
histogram_points_nine = []
for i in range(0, 128, 8):
temp = []
for j in range(0, 64, 8):
magnitude_values = [[mag[i][x] for x in range(j, j+8)] for i in range(i,i+8)]
angle_values = [[theta[i][x] for x in range(j, j+8)] for i in range(i, i+8)]
for k in range(len(magnitude_values)):
for l in range(len(magnitude_values[0])):
bins = [0.0 for _ in range(number_of_bins)]
value_j = calculate_j(angle_values[k][l])
Vj = calculate_value_j(magnitude_values[k][l], angle_values[k][l], value_j)
Vj_1 = magnitude_values[k][l] - Vj
bins[value_j]+=Vj
bins[value_j+1]+=Vj_1
bins = [round(x, 9) for x in bins]
temp.append(bins)
histogram_points_nine.append(temp)
print(len(histogram_points_nine))
print(len(histogram_points_nine[0]))
print(len(histogram_points_nine[0][0]))
>>>16
>>>8
>>>9
为由2x2的块提供HOG特征向量
(1个块由8x8个单元组成)
epsilon = 1e-05
feature_vectors = []
for i in range(0, len(histogram_points_nine) - 1, 1):
temp = []
for j in range(0, len(histogram_points_nine[0]) - 1, 1):
values = [[histogram_points_nine[i][x] for x in range(j, j+2)] for i in range(i, i+2)]
final_vector = []
for k in values:
for l in k:
for m in l:
final_vector.append(m)
k = round(math.sqrt(sum([pow(x, 2) for x in final_vector])), 9)
final_vector = [round(x/(k + epsilon), 9) for x in final_vector]
temp.append(final_vector)
feature_vectors.append(temp)
print(len(feature_vectors))
print(len(feature_vectors[0]))
print(len(feature_vectors[0][0]))
>>>15
>>>7
>>>36
获得的HOG特征数量
print(f'Number of HOG features = {len(feature_vectors) * len(feature_vectors[0]) * len(feature_vectors[0][0])}')
>>>Number of HOG features = 3780
在Python中使用skimage库的HOG特征
导入库
from skimage.io import imread
from skimage.transform import resize
from skimage.feature import hog
from skimage import exposure
import matplotlib.pyplot as plt
读取图像
img = imread('B.jpg')
plt.axis("off")
plt.imshow(img)
print(img.shape)
>>>(781, 794, 3)
缩放
resized_img = resize(img, (128*4, 64*4))
plt.axis("off")
plt.imshow(resized_img)
print(resized_img.shape)
>>>(512, 256, 3)
创建和可视化HOG特征
fd, hog_image = hog(resized_img, orientations=9, pixels_per_cell=(8, 8),
cells_per_block=(2, 2), visualize=True, multichannel=True)
plt.axis("off")
plt.imshow(hog_image, cmap="gray")
plt.show()
kaggle竞赛中的图像分割的Tips和Tricks
外部数据
- 使用 LUng Node Analysis Grand Challenge 数据,因为这个数据集包含了来自放射学的标注细节。
- 使用 LIDC-IDRI 数据,因为它具有找到了肿瘤的所有放射学的描述。
- 使用Flickr CC,维基百科通用数据集
- 使用Human Protein Atlas Dataset
- 使用IDRiD数据集
数据探索和直觉
- 使用0.5的阈值对3D分割进行聚类
- 确认在训练集和测试集的标签分布上有没有不一样的地方
预处理
- 使用DoG(Difference of Gaussian)方法进行blob检测,使用skimage中的方法。
- 使用基于patch的输入进行训练,为了减少训练时间。
- 使用cudf加载数据,不要用Pandas,因为读数据更快。
- 确保所有的图像具有相同的方向。
- 在进行直方图均衡化的时候,使用对比度限制。
- 使用OpenCV进行通用的图像预处理。
- 使用自动化主动学习,添加手工标注。
- 将所有的图像缩放成相同的分辨率,可以使用相同的模型来扫描不同的厚度。
- 将扫描图像归一化为3D的numpy数组。
- 对单张图像使用暗通道先验方法进行图像去雾。
- 将所有图像转化成Hounsfield单位(放射学中的概念)。
- 使用RGBY的匹配系数来找到冗余的图像。
- 开发一个采样器,让标签更加的均衡。
- 对测试图像打伪标签来提升分数。
- 将图像/Mask降采样到320x480。
- 直方图均衡化(CLAHE)的时候使用kernel size为32×32
- 将DCM转化为PNG。
- 当有冗余图像的时候,为每个图像计算md5 hash值。
数据增强
- 使用 albumentations 进行数据增强。
- 使用随机90度旋转。
- 使用水平翻转,上下翻转。
- 可以尝试较大的几何变换:弹性变换,仿射变换,样条仿射变换,枕形畸变。
- 使用随机HSV。
- 使用loss-less增强来进行泛化,防止有用的图像信息出现大的loss。
- 应用channel shuffling。
- 基于类别的频率进行数据增强。
- 使用高斯噪声。
- 对3D图像使用lossless重排来进行数据增强。
- 0到45度随机旋转。
- 从0.8到1.2随机缩放。
- 亮度变换。
- 随机变化hue和饱和度。
- 使用D4:https://en.wikipedia.org/wiki/Dihedral_group增强。
- 在进行直方图均衡化的时候使用对比度限制。
- 使用AutoAugment:https://arxiv.org/pdf/1805.09501.pdf增强策略。
模型
结构
- 使用U-net作为基础结构,并调整以适应3D的输入。
- 使用自动化主动学习并添加人工标注。
- 使用inception-ResNet v2 architecture结构使用不同的感受野训练特征。
- 使用Siamese networks进行对抗训练。
- 使用ResNet50, Xception, Inception ResNet v2 x 5,最后一层用全连接。
- 使用global max-pooling layer,无论什么输入尺寸,返回固定长度的输出。
- 使用stacked dilated convolutions。
- VoxelNet。
- 在LinkNet的跳跃连接中将相加替换为拼接和conv1x1。
- Generalized mean pooling。
- 使用224x224x3的输入,用Keras NASNetLarge从头训练模型。
- 使用3D卷积网络。
- 使用ResNet152作为预训练的特征提取器。
- 将ResNet的最后的全连接层替换为3个使用dropout的全连接层。
- 在decoder中使用转置卷积。
- 使用VGG作为基础结构。
- 使用C3D网络,使用adjusted receptive fields,在网络的最后使用64 unit bottleneck layer 。
- 使用带预训练权重的UNet类型的结构在8bit RGB输入图像上提升收敛性和二元分割的性能。
- 使用LinkNet,因为又快又省内存。
- MASKRCNN
- BN-Inception
- Fast Point R-CNN
- Seresnext
- UNet and Deeplabv3
- Faster RCNN
- SENet154
- ResNet152
- NASNet-A-Large
- EfficientNetB4
- ResNet101
- GAPNet
- PNASNet-5-Large
- Densenet121
- AC-GAN
- XceptionNet (96), XceptionNet (299), Inception v3 (139), InceptionResNet v2 (299), DenseNet121 (224)
- AlbuNet (resnet34) from ternausnets
- SpaceNet
- Resnet50 from selim_sef SpaceNet 4
- SCSEUnet (seresnext50) from selim_sef SpaceNet 4
- A custom Unet and Linknet architecture
- FPNetResNet50 (5 folds)
- FPNetResNet101 (5 folds)
- FPNetResNet101 (7 folds with different seeds)
- PANetDilatedResNet34 (4 folds)
- PANetResNet50 (4 folds)
- EMANetResNet101 (2 folds)
- RetinaNet
- Deformable R-FCN
- Deformable Relation Networks
硬件设置
- Use of the AWS GPU instance p2.xlarge with a NVIDIA K80 GPU
- Pascal Titan-X GPU
- Use of 8 TITAN X GPUs
- 6 GPUs: 21080Ti + 41080
- Server with 8×NVIDIA Tesla P40, 256 GB RAM and 28 CPU cores
- Intel Core i7 5930k, 2×1080, 64 GB of RAM, 2x512GB SSD, 3TB HDD
- GCP 1x P100, 8x CPU, 15 GB RAM, SSD or 2x P100, 16x CPU, 30 GB RAM
- NVIDIA Tesla P100 GPU with 16GB of RAM
- Intel Core i7 5930k, 2×1080, 64 GB of RAM, 2x512GB SSD, 3TB HDD
- 980Ti GPU, 2600k CPU, and 14GB RAM
损失函数
- Dice Coefficient ,因为在不均衡数据上工作很好。
- Weighted boundary loss 目的是减少预测的分割和ground truth之间的距离。
- MultiLabelSoftMarginLoss 使用one-versus-all损失优化多标签。
- Balanced cross entropy (BCE) with logit loss 通过系数来分配正负样本的权重。
- Lovasz 基于sub-modular损失的convex Lovasz扩展来直接优化平均IoU损失。
- FocalLoss + Lovasz 将Focal loss和Lovasz losses相加得到。
- Arc margin loss 通过添加margin来最大化人脸类别的可分性。
- Npairs loss 计算y_true 和 y_pred之间的npairs损失。
- 将BCE和Dice loss组合起来。
- LSEP – 一种成对的排序损失,处处平滑因此容易优化。
- Center loss 同时学习每个类别的特征中心,并对距离特征中心距离太远的样本进行惩罚。
- Ring Loss 对标准的损失函数进行了增强,如Softmax。
- Hard triplet loss 训练网络进行特征嵌入,最大化不同类别之间的特征的距离。
- 1 + BCE – Dice 包含了BCE和DICE损失再加1。
- Binary cross-entropy – log(dice) 二元交叉熵减去dice loss的log。
- BCE, dice和focal 损失的组合。
- BCE + DICE - Dice损失通过计算平滑的dice系数得到。
- Focal loss with Gamma 2 标准交叉熵损失的升级。
- BCE + DICE + Focal – 3种损失相加。
- Active Contour Loss 加入了面积和尺寸信息,并集成到深度学习模型中。
- 1024 * BCE(results, masks) + BCE(cls, cls_target)
- Focal + kappa – Kappa是一种用于多类别分类的损失,这里和Focal loss相加。
- ArcFaceLoss — 用于人脸识别的Additive Angular Margin Loss。
- soft Dice trained on positives only – 使用预测概率的Soft Dice。
- 2.7 * BCE(pred_mask, gt_mask) + 0.9 * DICE(pred_mask, gt_mask) + 0.1 * BCE(pred_empty, gt_empty) 一种自定义损失。
- nn.SmoothL1Loss()。
- 使用Mean Squared Error objective function,在某些场景下比二元交叉熵损失好。
训练技巧
- 尝试不同的学习率。
- 尝试不同的batch size。
- 使用SGD + 动量 并手工设计学习率策略。
- 太多的增强会降低准确率。
- 在图像上进行裁剪做训练,全尺寸图像做预测。
- 使用Keras的ReduceLROnPlateau()作为学习率策略。
- 不使用数据增强训练到平台期,然后对一些epochs使用软硬增强。
- 冻结除了最后一层外的所有层,使用1000张图像进行微调,作为第一步。
- 使用分类别采样
- 在调试最后一层的时候使用dropout和增强
- 使用伪标签来提高分数
- 使用Adam在plateau的时候衰减学习率
- 用SGD使用Cyclic学习率策略
- 如果验证损失持续2个epochs没有降低,将学习率进行衰减
- 将10个batches里的最差的batch进行重复训练
- 使用默认的UNET进行训练
- 对patch进行重叠,这样边缘像素被覆盖两次
- 超参数调试:训练时候的学习率,非极大值抑制以及推理时候的分数阈值
- 将低置信度得分的包围框去掉。
- 训练不同的卷积网络进行模型集成。
- 在F1score开始下降的时候就停止训练。
- 使用不同的学习率。
- 使用层叠的方法用5 folds的方法训练ANN,重复30次。
评估和验证
- 按类别非均匀的划分训练和测试集
- 当调试最后一层的时候,使用交叉验证来避免过拟合。
- 使用10折交叉验证集成来进行分类。
- 检测的时候使用5-10折交叉验证来集成。
集成方法
- 使用简单的投票方法进行集成
- 对于类别很多的模型使用LightGBM,使用原始特征。
- 对2层模型使用CatBoost。
- 使用 ‘curriculum learning’ 来加速模型训练,这种训练模式下,模型先在简单样本上训练,再在困难样本上训练。
- 使用ResNet50, InceptionV3, and InceptionResNetV2进行集成。
- 对物体检测使用集成。
- 对Mask RCNN, YOLOv3, 和Faster RCNN 进行集成。
后处理
- 使用test time augmentation ,对一张图像进行随机变换多次测试后对结果进行平均。
- 对测试的预测概率进行均衡化,而不是使用预测的类别。
- 对预测结果进行几何平均。
- 在推理的时候分块重叠,因为UNet对边缘区域的预测不是很好。
- 进行非极大值抑制和包围框的收缩。
- 在实例分割中使用分水岭算法后处理来分离物体。