
优化
weixin_39566143
这个作者很懒,什么都没留下…
展开
-
使用Opencv运行onnx模型-从训练到部署
1. 首先使用pytorch训练一个简单的猫狗分类模型具体代码参考pytorch实现kaggle猫狗识别(超详细)记得先下载猫狗数据集,然后改一下代码里的路径2. 将保存的模型转为onnx格式model = torch.load('model.pt',map_location=lambda storage, loc: storage)dummy = torch.randn(1,3,224,224)out = model(dummy)torch.onnx.export(model,dummy,原创 2021-07-09 11:21:44 · 6717 阅读 · 10 评论 -
一些加快python执行速度的小技巧
优化Python的执行速度1. 尽量不要使用全局变量,把代码块封装成函数,看下面的代码import time# 全局代码块start = time.perf_counter()for i in range(100000000): i += 1print('global cost time:',time.perf_counter()-start)def fun(): for j in range(100000000): j += 1# 函数局部代码块原创 2021-03-07 16:49:33 · 382 阅读 · 1 评论