- 博客(15)
- 资源 (1)
- 收藏
- 关注
RSS订阅原创 pytorch将模型转换为ONNX模型
使用torch.onnx.export()转换模型为onnximport torchimport torchvisionfrom utils import get_network #此处是自己定义的获取模型结构的模块net = get_network("mobilenet") #以Mobilenet为例weight = "***.pth" #只保存了模型参数net.load_state_dict(torch.load(weight))#若.pth文件保存了模型结构和参数，
2021-08-06 10:00:22
633
原创 将创建的虚拟环境导入jupyter
在终端下：conda create -n py python=3 #创建一个虚拟环境conda activate py # 进入虚拟环境conda install ipykernel #安装ipykernel模块python -m ipykernel install --user --name py(此处是自己要导入jupyter的虚拟环境名字) --display-name "py3"(此处是虚拟环境在jupyter的显示名字)jupyter notebook #最后启动jupyter
2020-10-15 10:27:29
505
原创 Pytorch训练相关
define optimizer针对需要训练特定参数定义训练器params = [p for p in model.parameters() if p.requires_grad]optimizer = torch.optim.SGD(params, lr=0.004, momentum=0.9, weight_decay=0.0005)Save weightssave_files = {'model' : model.state_dict(),'optimizer' : optimizer
2020-10-11 16:18:41
257
原创 python视频抽帧保存为视频
视频抽帧保存为视频"""@Author:yebin@file_name:视频扩充.py@Time:2020/8/21 上午11:10@function:"""import cv2import osimport globdef getPath(rootPath): video_paths = glob.glob(rootPath + '/*.mp4') return video_pathsdef readVideo(video_path): img
2020-09-11 09:57:49
753
2
原创 tensorflow学习笔记
1.tf.InteractiveSession() 与tf.Session()的区别：tf.InteractiveSession():它能让你在运行图的时候，插入一些计算图，这些计算图是由某些操作(operations)构成的。这对于工作在交互式环境中的人们来说非常便利，比如使用IPython。tf.Session():需要在启动session之前构建整个计算图，然后启动该计算图。意思就是在我们使用tf.InteractiveSession()来构建会话的时候，我们可以先构建一个session然后再定
2020-08-01 17:35:30
121
原创 2020-07-31
????安全AI挑战者计划第四期评测开始啦！报名拼速度！数据+白盒模型定义、权重已经可以通过官网下载到，使用示例中文版：https://tianchi.aliyun.com/forum/postDetail?spm=5176.12586969.1002.3.5ca36163tB6GJd&postId=117459英文版：https://tianchi.aliyun.com/forum/postDetail?spm=5176.12586969.1002.6.5ca36163tB6GJd&
2020-07-31 11:38:27
100
原创 ubuntu内存使用查看及清理内存
首先在终端执行一下命令查看内存使用情况free -m清理内存命令：sudo sh -c 'echo 3 > /proc/sys/vm/drop_caches'再次执行命令free -m 查看空闲内存
2020-07-28 11:43:17
1171
原创 C++
//用于裁剪视频#include <opencv2/core/core.hpp>#include <opencv2/highgui/highgui.hpp>#include <opencv2/imgproc/imgproc.hpp>using namespace std;using namespace cv;int main(){ // 截取小视频// std::string srcVideoPath = "/data_2/projec
2020-07-21 18:00:32
144
原创 C++实现视频帧获取并跳帧保存
每四帧保存//// Created by em on 20-7-21.//Author: Bumblebee#include <opencv2/core/core.hpp>#include <opencv2/highgui/highgui.hpp>#include <opencv2/imgproc/imgproc.hpp>#include <iostream>using namespace std;using namespace cv;
2020-07-21 16:23:33
2129
原创 划分数据集
import globimport shutiltotal_file = 'total_file.txt'train = 'train.txt'val = 'val.txt'bin_path = '点云文件路径' # 总的点云数据集文件路径bin_train_dir = '训练集保存路径'bin_val_dir = '验证集保存路径'total_bin = glob.glob(bin_path)split_rate = 0.5total_bin_num = len(
2020-07-16 11:58:31
209
原创 python将指定文件移动到指定目录
将 f1移动到 f2目录下需要导入 shutil模块#将文件f1 移动到 f2目录下import shutilf1 = '../anchors/basline_anchors.txt'f2 = './'shutil.move(f1, f2)
2020-07-14 15:36:25
5341
1
原创 Python从键盘中输入一个列表
从键盘上输入一个列表，在键盘上输入形式：[1,2,3,5,6]直接上代码x = input()x = x.strip('[')x = x.strip(']')xlist = x.split(',')xlist = [int(xlist[i]) for i in range(len(xlist))]print(xlist)从键盘键入[1,2,3,5,6]运行结果：[1, 2, 3, 5, 6]Process finished with exit code 0...
2020-07-10 14:26:05
17264
原创 tf.Variable() trainable的理解
看了网上的一些解释：如果为True，则会默认将变量添加到图形集合GraphKeys.TRAINABLE_VARIABLES中。此集合用于优化器Optimizer类优化的的默认变量列表，如果为False则在训练时不会更新该值。自己实验了一下：import tensorflow as tfx = tf.Variable(1.0, dtype=tf.float64, trainable=False, name='global_step')y = tf.Variable(2.0, dtype=tf.fl
2020-07-09 10:03:54
1733
1
原创 在Ubuntu导入Tesorflow出错问题
文章目录#1.ImportError: libcusolver.so.10.0: cannot open shared object file: No such file…分析：可能是cudnn未安装执行命令查看cudnn是否安装cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2若有结果输出则表示cudnn已安装的，若没有结果，则需要进行cudnn安装首先先查看自己的tensorflow，cuda，cudnn的适配版本，此处可
2020-07-07 14:16:03
112
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人