- 博客(17)
- 收藏
- 关注
原创 SOTA到底是什么算法
SOTA到底是什么算法?最近在看论文时看到了SOTA模型,以为是个很牛逼的算法,找了半天原理SOTA是state of the art的缩写。SOTA model:state-of-the-art model,并不是特指某个具体的模型,而是指在该项研究任务中,目前最好/最先进的模型。SOTA result:state-of-the-art result,指的是在该项研究任务中,目前最好的模型的结果/性能/表现。...
2021-06-16 14:25:28
32643
15
原创 TensorRT学习资料
最近准备学习一下TensorRT,发现了一个很详细的博客资料,记录之,方便以后学习https://www.cnblogs.com/vh-pg/p/11677137.html
2020-12-21 13:51:53
381
原创 ubuntu下pycharm创建快捷方式
在ubuntu下安装pycharm之后会发现没有桌面快捷方式。想要快捷方式需要在安装后做以下操作:打开termianl,输入:sudo gedit /usr/share/applications/Pycharm.desktop输入完成后会进入文本编辑界面,在打开的文本中输入:[Desktop Entry]Type=ApplicationName=PycharmGenericName=Pycharm3Comment=Pycharm3:The Python IDEExec=sh /hom
2020-09-01 21:40:22
288
原创 pytorch预训练下载保存路径
pytorch下载的预训练参数保存路径:/.cache/torch/hub/checkpoints或者/.cache/torch/checkpoints
2020-08-31 16:45:35
1763
原创 c++ 字符串char[]数组分割split
使用c++提供的strtok进行字符串数组分割,话不多说,上代码:#include <iostream>#include <string>#include <vector>using namespace std;int main(){ vector<char*> res_split;//存储分割后的字符串 char *name = "camera 10_2020821_1_23.bmp"; char image_name[
2020-08-28 16:30:35
8127
2
原创 pytorch 1.6.0 出现ModuleNotFoundError: No module named ‘tensorboard‘错误
pip install tb-nightly
2020-08-25 11:13:52
6938
1
原创 多目标图像分割中,每个目标分割精度计算
最近的项目中需要评价图像分割中每个目标的分割精度,如果目标分割精度大于阈值,则分割成功。每个图像中存在多个目标,分割结束后,如何评价每个目标的分割效果。本文的做法是crop出每个目标,分别计算其iou。流程:1、读取mask与resut。2、对于mask使用opencv中的findcontours函数,找出每个目标。3、用boundingRect求出目标的外接矩形,并crop出,与result对比计算。iou计算:def iou(mask, result): intersection =
2020-08-24 16:13:51
1718
2
原创 清华镜像安装pytorch
pip install torch torchvision -i https://pypi.tuna.tsinghua.edu.cn/simple
2020-08-23 17:22:07
650
1
原创 labelme json转换的label.png读取像素值全为0的问题
最近在图像分割时用到labelme进行图像标注。将标注好的json转换为label.png,用opencv读取图像时发现全为0。用PIL.Image读能读出标注的label。后来发现opencv在读图像时有几种选择:import cv2cv2.imread("filename",flags)IMREAD_UNCHANGED = -1#不进行转化,比如保存为了16位的图片,读取出来仍然为16位。IMREAD_GRAYSCALE = 0#进行转化为灰度图,比如保存为了16位的图片,读取出来为8位,类型
2020-08-19 09:39:21
1655
7
原创 对vector内存储的Point进行排序
#include <iostream>#include <opencv2/opencv.hpp>using namespace std;using namespace cv;#定义排序规则bool comp(Point a, Point b){ return a.x>b.x;}int main(){ vector<Point> p; p.push_back(Point(1, 2)); p.push_bck(Poi
2020-08-12 09:32:08
2585
原创 windows 10 c++(libtorch)调用pytorch训练好的分割模型
c++调用训练好的分割模型主要步骤如下:1、将训练好的.pth 模型转换为.pt模型2、opencv读取要推理的图像3、model前向传播,获得推理结果4、将推理结果转为opencv,查看推理结果。开发环境windows10vs 2015pytorch 1.2.0 cpuopencv 4.3.0libtorch 1.3.0 cpu配置开发环境时遇到的坑:1、vs2015不支持最新的libtorch2、ptorch与liborch 使用的cpu或者gpu版本要统一,否则出现如下错误
2020-08-10 11:54:11
4255
27
原创 pytorch训练好的.pth模型转换为.pt
将python训练好的.pth文件转为.ptimport torchimport torchvisionfrom unet import UNetmodel = UNet(3, 2)model.load_state_dict(torch.load("best_weights.pth"))model.eval()example = torch.rand(1, 3, 320, 480)traced_script_module = torch.jit.trace(model, example)t
2020-08-06 15:28:43
16748
17
原创 pytorch 对image和label同时进行随机翻转
在使用pytorch进行数据增广时碰到一个坑,对image和label分别进行增广,网络训练结果一直很差,查了很久才找到,原来分别image和label进行随机翻转操作时,是不同步的,记录之,防止以后再犯。所以在对数据进行随机增广操作时,需要指定的参数,代码如下:image_path = '/home/org/19.bmp'label_path = '/home/mask/19.png'image = Image.open(image_path)print(image.size)label =
2020-08-06 12:02:42
5384
2
原创 tensorflow2.x运行tensorflow1.x的代码
tensorflow2.x兼容tensorflow1.x版本,分为两布:1、import tensorflow.compat.v1 as tf 代替 import tensorflow as tf2、执行tf.disable_eager_execution()代码实例:# import tensorflow as tfimport tensorflow.compat.v1 as tftf.disable_eager_execution()v1 = tf.constant([1, 2, 3,
2020-08-04 21:00:24
1422
原创 ## ReadTimeoutError:HTTPSConnectionPool(host=‘files.pythonhosted.org‘, port=443): Read timed out.
pip install 出现如标题所示的错误时,使用如下代码,亲测有效:pip install *** --default-timeout=1000
2020-08-03 22:12:55
267
原创 Faster-RCNN预训练结果
Faster-RCNN预训练结果最近用到TensorRT,需要运行官方给出的faster-rcnn的sample,但是readme中给出的预训练模型下载链接不能访问。花了很久才找到,现在给出预训练模型的百度网盘链接,有需要的自取。链接中有两个文件分别为VGG16_faster_rcnn_final.caffemodel、ZF_faster_rcnn_final.caffemodel。tensort官方faster-rcnn sample中用到的为第一个。链接:https://pan.baidu.com/
2020-07-07 14:19:27
839
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人