numpy
York1996
懒人一枚,热爱提高效率,简化流程,自动办公;
展开
-
np.argmax没有keepdims得可能原因
可能是因为您的numpy版本太老了,1.22版本以上的才有keepdims参数。而1.22以上的,至少要3.8以上的python版本。原创 2023-06-13 11:05:54 · 264 阅读 · 0 评论 -
cv2.fillPoly报错的可能原因
其中,pts的数据类型应该是np.int32,或者64,而不是np.uint8。原创 2022-08-08 11:31:16 · 1885 阅读 · 0 评论 -
np.sum多维度求和求均值等 TypeError: ‘list‘ object cannot be interpreted as an integer的可能原因
会报错,axis应该写成元组的格式。得到结果[10000. 10000. 10000.]原创 2022-08-05 15:38:26 · 335 阅读 · 0 评论 -
pytorch转caffe遇到的问题、经验总结;
转换的时候,需要写一个caffe的proto,还好我之前是根据caffe模型写的pytorch模型,所以变量的命名基本一致,所以转换的时候做对应就简单一些。设计网络的时候,最好用流程图画出网络模型,然后在开始写代码。提取pth中模型参数的时候,不需要原始的网络,pth使用torch读取之后,就已经是一个dict结构了。这个参数赋值给caffe的时候,需要caffe和torch权重的对应关系,然后一一赋值即可。做好之后,将结果可视化出来,看看直观上看上去是否完全一致。最后如果有测试集的话,测试原创 2021-12-13 15:54:55 · 1025 阅读 · 0 评论 -
python3保存的npy文件,python2无法读取的可能解决方案。
使用h5py模块来存数据python3保存文件:import h5pywith h5py.File('weight.h5','w') as f: for key in state_dict: value=state_dict[key].cpu().numpy() f[key]=value print(key)r然后用python2读取:with h5py.File(model_path, 'r') as f:a=...原创 2021-12-13 14:02:14 · 1478 阅读 · 0 评论 -
no mudule named six的可能解决方法
用conda安装sixconda install six原创 2021-12-02 21:19:33 · 342 阅读 · 0 评论 -
np.diff二维数组中使用append和prepend,
以下代码实现了在a的第一列重复一次,再执行diff操作。a=np.array([ [1,2,3], [4,5,6], [4, 5, 6], [7,8,9], [7, 8, 9], [1, 8, 4],])print(np.diff(a,axis=1,prepend=a[:,0:1]))结果:[[ 0 1 1][ 0 1 1][ 0 1 1][ 0 1 1][ 0 1 1][ 0 7 -4]...原创 2020-06-27 18:45:54 · 1794 阅读 · 0 评论