随笔分类 -  pytorch

记录使用pytorch实现的算法
pytorch resnet18转换为ncnn
摘要：模型简化： python3 -m onnxsim input_onnx_model output_onnx_model 转换完成后，需要注意： 1.图像归一化方式： 如果训练过程中，图像归一化为： preprocess_transform = transforms.Compose([ transfo 阅读全文
posted @ 2020-12-18 18:31 快乐码小农 阅读(899) 评论(0) 推荐(0)
siamRPN网络结构理解
摘要：siamRPN包含两个分支。 z->exemplar(1,3,127,127), x->instance(1,3,255,255) **branch1:** 对z提取特征fz(AlexNet)，并根据f计算分类和回归卷积核 fz shape->(1,256,6,6) 分类cls: fcls(fz)- 阅读全文
posted @ 2020-11-25 17:47 快乐码小农 阅读(311) 评论(0) 推荐(0)