- 博客(7)
- 收藏
- 关注
原创 Pytorch模型转TorchScript及部署
Pytorch模型转TorchScript1.背景2. 转换方法Pytorch框架TorchScript模型转换方法4. 部署#include<opencv2/opencv.hpp>#include <torch/torch.h>#include <torch/script.h> int main(){ //定义使用cuda auto device = torch::Device(torch::kCUDA,0); //读取图片 auto ima
2022-05-13 17:01:52
1547
原创 西电研究生毕业论文latex模板使用指南
西电研究生毕业论文latex模板使用指南记录西电研究生毕业论文latex模板过程中一些需要更改的地方下载学校提供的latex模板后,在winEdt7.0中运行templet.tex,根据提示安装需要的sty包。一些学长学姐的排版参考:https://blog.youkuaiyun.com/niaolianjiulin/article/details/79269490https://blog.youkuaiyun.com/niaolianjiulin/article/details/79461615https:/
2021-04-10 21:36:05
1053
原创 语义分割的评价指标(OA, F1 score)计算(不使用sklearn)
混淆矩阵(confusion_matrix)是由预测值和标签值组成的二维矩阵,共n行n列。n表示class类别数。二分类图片源自https://blog.youkuaiyun.com/Orange_Spotty_Cat/article/details/80520839多分类#计算 overall accuracyoa = np.diag(conf_mat).sum() / conf_mat.sum()#计算各类别 accuracyacc_cls = np.aidg(conf_mat) / conf_m
2021-01-11 21:27:11
4184
1
原创 Tensorflow训练过程中validation
Tensorflow训练过程中validation载入数据定义is_training占位符用is_training控制图结点唯一sess运行Tensorflow因为静态图的原因,边train边validation的过程相较于pytorch来说复杂一些。载入数据分别获取训练集和验证集的数据。我这里使用的是从tfrecoed读入数据。# training dataimg_name_batch_train, img_batch_train, gtboxes_and_label_batch_train,
2021-01-08 20:35:05
1181
原创 python用sys.stdin和sys.stdout输入输出——牛客笔试常用!!!
stdin 输入需要import sys1.sys.stdin.readline().strip()读一行输入,默认输入是字符串。strip()会将标准输入末尾的'\n'去除。2. sys.stdin.readline().strip().split()读一行输入,行中每个字符以空格间隔,若是逗号间隔,使用sys.stdin.readline().strip().split(‘,’)。3. in =sys.stdin.readline().strip()list(map(in..
2020-06-15 20:10:46
1294
转载 tf.shape()与tensor.get_shape()
tensorflow tf.shape()与tensor.get_shape()相同點两种方法都可以获得变量的shape。不同點tf.shape(x):返回tensor。如果在运行的时候想知道None到底是多少,这时候,只能通过tf.shape(x)[0]讀取。tensor.get_shape():返回tuple。無法讀取None。...
2020-05-23 21:35:00
313
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人