工作记录
小关学长
这个作者很懒,什么都没留下…
展开
-
利用python获取windows管理员权限执行代码
import ctypes, sysimport osdef is_admin(): try: return ctypes.windll.shell32.IsUserAnAdmin() except: return Falseif is_admin(): # 这里写入需要管理员权限执行的操作 passelse: if...原创 2019-11-23 21:05:48 · 5154 阅读 · 1 评论 -
pandas去除某几列相同数据的记录并保存
1 问题现在有这样一个需求,在一份记录着微信聊天记录的 csv 文件中,去除同一天同一个人发的同一条消息的记录。2 实现代码# -*- coding: utf-8 -*-import pandas as pdimport osfrom tqdm import *class DeleteRepetition: def __init__(self, finished_filed...原创 2019-04-08 17:41:44 · 3600 阅读 · 0 评论 -
TensorRT5介绍及Pytorch转TensorRT5代码示例
1 TensorRT简介TensorRT的核心是一个c++库,它促进了对NVIDIA图形处理单元(gpu)的高性能计算。它与TensorFlow,Pytorch等框架相辅相成。他可以快速高效的运行一个已经训练好的神经网络,并生成结果。它包括用于从Caffe、ONNX或TensorFlow导入现有模型的解析器,以及用于以编程方式构建模型的c++和Python api。TensorRT在所有支持的平...原创 2019-04-12 09:33:54 · 10423 阅读 · 3 评论 -
Pandas从原dataframe中剔除筛选出来的dataframe
# 随机取一定数量的就splitfile_one_df = csvfile.sample(num)# 源文件剔除随机到的数据splitfile_two_df = csvfile.drop(splitfile_one_df.index)原创 2019-04-12 14:42:11 · 5240 阅读 · 0 评论 -
pytorch转onnx模型多输入问题(如:Bert)
举个例子:Bert模型有三个输入,因此就要创建三个dummy_input,然后利用一个tuple,传入函数中。dummy_input0 = torch.LongTensor(Batch_size, seg_length).to(torch.device("cuda"))dummy_input1 = torch.LongTensor(Batch_size, seg_length).to(tor...原创 2019-04-26 12:21:57 · 14914 阅读 · 4 评论 -
Pytorch通过保存为ONNX模型转TensorRT5
1 Pytorch以ONNX方式保存模型 def saveONNX(model, filepath): ''' 保存ONNX模型 :param model: 神经网络模型 :param filepath: 文件保存路径 ''' # 神经网络输入数据类型 dumm...原创 2019-04-15 16:08:35 · 17610 阅读 · 28 评论