2022年8月28日
pytorch模型转mindspore模型的相似度
摘要： test_image = 'images/hand.jpg' oriImg = cv2.imread(test_image) # B,G,R order w=body_estimation(oriImg)#直接存w0就行。 np.save('input.npy',oriImg) np.save('o 阅读全文
posted @ 2022-08-28 17:28 祥瑞哈哈哈 阅读(99) 评论(0) 推荐(0)
判断俩个numpy相差大小
摘要： import numpy as np out1=np.load('output1.npy') out2=np.load('output2.npy') t1=np.load('t1.npy') t2=np.load('t2.npy') result = np.absolute(np.array(out 阅读全文
posted @ 2022-08-28 17:24 祥瑞哈哈哈 阅读(46) 评论(0) 推荐(0)
pytorch转为mindspore模型
摘要： MindConverter将PyTorch(ONNX)模型快速迁移到MindSpore框架下使用。 第一步:pytorch模型转onnx: import torch # 根据实际情况替换以下类路径 from customized.path.to.pytorch.model import PyTorc 阅读全文
posted @ 2022-08-28 11:25 祥瑞哈哈哈 阅读(1169) 评论(0) 推荐(0)