a819411321的博客

公司最近做异常检测，用One-Class做，感觉很棒，有python代码有c++部署代码。所需要做得就是如何提高训练效果。先鸽一会，下周写，坑站着。One Class是一种使用gan的缺陷检测方案，该方案仅依靠正样本和少量的负样本即可实现缺陷检测。训练集合的样本是生产出来的物品，标签是标准的物品。样本标签样本经过CNN（解码网络）得到Feature map 然后再经...

先贴网络吧我所用的One-class数据和网络如下：#以256*64的数据为例子，grid为4，batch为4。#在训练网络权重时，数据经过每一层后的维度如下：torch.Size([4, 3, 64, 256]) torch.Size([4, 48, 32, 128])torch.Size([4, 96, 33, 64])torc...

One-class算法有编码和解码两个过程。在解码阶段，需要将feature map 重新变成图像。One-class算法用的上采样方法是ConvTranspose2d，公司想对One-class进行改进，就让我调研可不可以将ConvTranspose2d用其他的上采样方法替换。我就去调研了一下，发现上采样方法有这几种。上采样方法介绍：1.转置卷积2.Pixelshuffle3....