python
通过幸福的路唯有奋斗
一往无前,努力,奋斗!
展开
-
目标跟踪算法IOU Tracker与SORT算法
重新写了在线版本的IOU Tracker代码,并封装成类。可实时运行检测的同时进行跟踪。原始sort只要生成tracker就会分配编号,现在修改未只有连续检测并跟踪到N帧图像才会被判定为有效tracker,这时候才会分配跟踪id。原创 2022-11-17 11:19:01 · 774 阅读 · 1 评论 -
PyTorch 保存与加载模型
保存模型与加载模型常用的两种保存与加载模型方式# 1.保存整个网络torch.save(net, PATH)#针对上面保存方法,加载的方法是:model_dict=torch.load(PATH)# 2.保存网络中的参数, 速度快,占空间少torch.save(net.state_dict(),PATH)#针对上面的保存方法,加载的方法是:model_dict=model.load_state_dict(torch.load(PATH))...原创 2020-08-06 10:52:32 · 182 阅读 · 0 评论 -
np.pad()用法
numpy中pad函数的常用方法一、参数解释ndarray = numpy.pad(array, pad_width, mode, **kwargs)array为要填补的数组 pad_width是在各维度的各个方向上想要填补的长度,如((1,2),(2,2)),表示在第一个维度上水平方向上padding=1,垂直方向上padding=2,在第二个维度上水平方向上padding=2,...转载 2020-04-15 17:19:27 · 2069 阅读 · 0 评论 -
python 定位array坐标
python中快速寻找array坐标索引import numpy as npa = np.array([[0,0,1,2], [1,1,0,0]])# 找到a内大于0的坐标并返回为arraycoords = np.array(np.where(a>0))# 返回值array([[0,0,1,1], [2,3,0,1]])...原创 2020-03-24 10:14:29 · 2991 阅读 · 0 评论 -
python 两个list元素求或
python 两个list元素求或,同理也可进行其他操作a = [1,0,0,1]b = [0,1,1,0]c = [a[i] or b[i] for i in range(len(a))]print cc = [1,1,1,1]原创 2019-08-27 15:38:25 · 1011 阅读 · 0 评论