- 博客(15)
- 资源 (10)
- 问答 (1)
- 收藏
- 关注
RSS订阅原创 libtorch yolov8 seg
https://github.com/a819411321/liborch-yolov8-seg-喜欢的点个赞
2024-03-12 16:44:28
822
4
原创 libtorch转换细节
Pytorch到libtorch存在BN层参数裁剪楼主有两个模型，两个模型都是AE。一个是用非全零图片训练的AE，一个是用大部分全零图片训练的。非全零图片训练的模样，pytorch推导结果和libtorch结果一致。大部分全零图片的AE模型的libtorch结果和pytoch结果不一致。当去除AE网络的BN层后，发现结果又一致了。猜测torch.jit.trace会裁剪bn参数。...
2019-12-07 09:08:45
594
4
原创 基于Oneclass的上采样方法对比
One-class算法有编码和解码两个过程。在解码阶段，需要将feature map 重新变成图像。One-class算法用的上采样方法是ConvTranspose2d，公司想对One-class进行改进，就让我调研可不可以将ConvTranspose2d用其他的上采样方法替换。我就去调研了一下，发现上采样方法有这几种。上采样方法介绍：1.转置卷积2.Pixelshuffle3....
2019-09-03 16:21:19
502
原创 One-class
先贴网络吧我所用的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...
2019-09-02 09:47:19
867
1
原创 One-Class Classification入门
公司最近做异常检测，用One-Class做，感觉很棒，有python代码有c++部署代码。所需要做得就是如何提高训练效果。先鸽一会，下周写，坑站着。One Class是一种使用gan的缺陷检测方案，该方案仅依靠正样本和少量的负样本即可实现缺陷检测。训练集合的样本是生产出来的物品，标签是标准的物品。样本标签样本经过CNN（解码网络）得到Feature map 然后再经...
2019-07-26 10:38:00
8424
7
原创 python多线程
"""TRAIN GANOMALY. Example: Run the following command from the terminal. run train.py \ --model ganomaly \ --dataset UCSD_Ano...
2019-07-26 10:25:20
171
原创 libtorch模型生成
import torchimport torch.nn as nnimport numpy as npimport torchvisionimport cv2from collections import OrderedDict#我们所用的解码网络class Encoder(nn.Module): def __init__(self, isize=256, nc=3, n...
2019-07-26 10:17:27
3368
1
原创 YoloV3的Pytorch工作总结
pytorch代码参考githubhttps://github.com/eriklindernoren/PyTorch-YOLOv3部署方法使用libtorch1.0+opencv3 不建议使用libtorch1.1另外libtorch 本人试过cpu版本和cuda9.0的gpu的版本 在windows和linux下均跑通过linux下的编译文件写法cmake_minimum_r...
2019-07-26 09:53:09
1704
8
原创 workday-4~5 win10下vs2015+opencv2+libtorch实现yolov3模型部署
阶段总结 libtorch实现yolov3模型部署 libtorch只使用了官方的编译具体方法：等有时间在慢慢详细写 今天先写需要环境。win10，vs2015，torch.1.0 opencv2.4 libtorch1.0先上效果...
2019-07-26 09:40:00
285
2
原创 workday-4 libtorch在windows下实现
首先感谢https://zhuanlan.zhihu.com/p/52806730搞了一下午终于在win10下的vs2015实现了简单的libtorch功能，接下来看看模型怎么部署。说一下我实现的过程。说一下我电脑的所有环境，win10 （64位）cud9.0 cudnn7.1按上面配置去官网下载torch1.0和libtorchtorch不说说一下libtorch的怎么在...
2019-07-26 09:38:30
222
原创 workday-1~3（我的工作日记）
进入某视觉公司，领了笔记本，电脑显卡是MX150。安装了显卡驱动390.87，以及cuda9和cudnn7.1。系统ubuntu16.04，nvidia-smi有显示，nvcc -V也有显示（成功显示9.0版本） 。然后安装tensorflow-gpu==1.8，成功后import 报错。。。。。。安装pytorch，import报错。无奈都装了cpu版本。（两天任务）深深怀疑MX150不能做...
2019-07-26 09:38:23
297
原创 YoloV3 最为详细的实现过程
最近要去工作了，好久没做图像检测了所以跑了跑代码，熟悉熟悉。跑的YoloV3 tensorflow代码从是github的https://github.com/YunYang1994/tensorflow-yolov3下载的 感谢大佬实现过程如下所用代码和数据集后续上传 审核过不过看天了代码下载https://download.youkuaiyun.com/download/a819411321...
2019-07-26 09:37:53
1967
检测矩形框（手机）改进版
2018-01-19
五点平滑算法轨迹优化
2018-08-16
安全帽人脸二分类的训练和测试代码、数据集和训好的模型.zip
2020-06-23
qt_bilibili.rar
2021-02-02
检测边框的 C++
2017-11-22
LSTM输入数据格式问题
2018-11-05
TA创建的收藏夹 TA关注的收藏夹
TA关注的人