yolov3（pytorch）转 ncnn 详解 |【❤️yolov3-spp.pt 转 onnx 转 ncnn❤️】
【摘要】
yolov3-spp.pt 【 pytorch 】模型 转 onnx 转 ncnn 推理
yolov3-spp.pt ==>> yolov3-spp.onnx ==>&...
yolov3-spp.pt 【 pytorch 】模型 转 onnx 转 ncnn 推理
yolov3-spp.pt ==>> yolov3-spp.onnx ==>> yolov3_spp_pt.param 【这一步卡壳，暂无方案】
文章目录
思路一【使用 yolov5 下的 yolov5-5.0/models/export.py 】转 onnx
# 下载 yolov5-5.0 版本代码
cd yolov5-5.0
# 把 yolov3 模型拷贝到 该目录下
ll yolov3weights/
total 492784
drwxrwxr-x 2 zql zql 4096 Sep 24 07:06 ./
drwxrwxr-x 10 zql zql 4096 Sep 24 07:05 ../
-rw-rw-r-- 1 zql zql 252297587 Sep 24 07:05 yolov3-spp.pt
-rw-rw-r-- 1 zql zql 252297867 Sep 24 07:05 yolov3-spp-ultralytics.pt
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
遇到
AttributeError报错如下 【暂不处理】
# 运行命令
python models/export.py --weights ./yolov3weights/yolov3-spp.pt --img 608 --batch 1
# 输出如下
Namespace(batch_size=1, device='cpu', dynamic=False, grid=False, img_size=[608, 608], weights='./yolov3weights/yolov3-spp.pt')
YOLOv5 🚀 2021-4-11 torch 1.9.0+cu102 CPU
Traceback (most recent call last):
File "models/export.py", line 51, in <module>
model = attempt_load(opt.weights, map_location=device) # load FP32 model
File "/home/zql/project/project21Next/modelTrans/ncnnLearn/yolov5-5.0/models/experimental.py", line 133, in attempt_load
model.append(ckpt['ema' if ckpt.get('ema') else 'model'].float().fuse(
AttributeError: 'collections.OrderedDict' object has no attribute 'float'
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
实际采用，yolov3 仓库 export.py 中的 ONNX_EXPORT
所使用代码
准备工作
1 修改
ONNX_EXPORT = True
2 下载 yolov3 pytorch 模型到 yolov3/weights 目录
yolov3-spp.pt 转 yolov3-spp.onnx
weights/ 目录下生成得到 yolov3-spp.onnx
python3 detect.py --source data/samples/ --weights weights/yolov3-spp.pt
# 运行输出如下
Namespace(agnostic_nms=False, augment=False, cfg='cfg/yolov3-spp.cfg', classes=None, conf_thres=0.3, device='', fourcc='mp4v', half=False, img_size=512, iou_thres=0.6, names='data/coco.names', output='output', save_txt=False, source='data/samples/', view_img=False, weights='weights/yolov3-spp.pt')
Using CPU
Fusing layers...
This means that the trace might not generalize to other inputs!
if nx == na: # same shape
graph torch-jit-export (
%images[FLOAT, 1x3x320x192]
) initializers (
%465[INT64, 2]
%466[INT64, 2]
%module_list.0.0.bias[FLOAT, 32]
%module_list.0.0.weight[FLOAT, 32x3x3x3]
...
...
%classes = Concat[axis = 0](%332, %395, %458)
%462 = Concat[axis = 0](%334, %397, %460)
%463 = Concat[axis = 0](%319, %382, %445)
%boxes = Concat[axis = 1](%462, %463)
return %classes, %boxes
}
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
yolov3-spp.onnx 转 ncnn【全过程】
onnxsim 对 yolov3-spp.onnx 进行 simplifier 处理得到 yolov3_spp_pt_sim.onnx
conda activate torch18
python -m onnxsim yolov3-spp.onnx yolov3_spp_pt_sim.onnx
# 输出如下
Simplifying...
Checking 0/3...
Checking 1/3...
Checking 2/3...
Ok!
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
yolov3_spp_pt_sim.onnx 转 ncnn
# 拷贝 刚刚的 yolov3_spp_pt_sim.onnx 到 如下目录
cd ncnn/build/tools/onnx
./onnx2ncnn yolov3_spp_pt_sim.onnx yolov3_spp_pt.param yolov3_spp_pt.bin
# 生成得到
ncnn.bin 和 ncnn.param
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
对 yolov3_spp_pt.param 进行修正 【卡住 - 暂无方案】
生成得到的 yolov3_spp_pt.param 如下，结构类似于 yolov5s ； 目前没有明确的 yolov3 算子替换案例；
因此，该步骤先进行到这里 | 【 大佬多多指教 】
7767517
270 309
Input images 0 1 images
MemoryData 310 0 1 310 0=2 1=180
MemoryData 318 0 1 318 0=2 1=180
MemoryData 333 0 1 333 0=2 1=180
MemoryData 373 0 1 373 0=2 1=720
MemoryData 381 0 1 381 0=2 1=720
MemoryData 396 0 1 396 0=2 1=720
MemoryData 436 0 1 436 0=2 1=2880
MemoryData 444 0 1 444 0=2 1=2880
MemoryData 459 0 1 459 0=2 1=2880
Convolution Conv_0 1 1 images 153 0=32 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=864
ReLU LeakyRelu_1 1 1 153 154 0=1.000000e-01
Convolution Conv_2 1 1 154 155 0=64 1=3 11=3 2=1 12=1 3=2 13=2 4=1 14=1 15=1 16=1 5=1 6=18432
ReLU LeakyRelu_3 1 1 155 156 0=1.000000e-01
Split splitncnn_0 1 2 156 156_splitncnn_0 156_splitncnn_1
Convolution Conv_4 1 1 156_splitncnn_1 157 0=32 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=2048
ReLU LeakyRelu_5 1 1 157 158 0=1.000000e-01
Convolution Conv_6 1 1 158 159 0=64 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=18432
ReLU LeakyRelu_7 1 1 159 160 0=1.000000e-01
BinaryOp Add_8 2 1 160 156_splitncnn_0 161 0=0
Convolution Conv_9 1 1 161 162 0=128 1=3 11=3 2=1 12=1 3=2 13=2 4=1 14=1 15=1 16=1 5=1 6=73728
ReLU LeakyRelu_10 1 1 162 163 0=1.000000e-01
Split splitncnn_1 1 2 163 163_splitncnn_0 163_splitncnn_1
Convolution Conv_11 1 1 163_splitncnn_1 164 0=64 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=8192
ReLU LeakyRelu_12 1 1 164 165 0=1.000000e-01
Convolution Conv_13 1 1 165 166 0=128 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=73728
ReLU LeakyRelu_14 1 1 166 167 0=1.000000e-01
BinaryOp Add_15 2 1 167 163_splitncnn_0 168 0=0
Split splitncnn_2 1 2 168 168_splitncnn_0 168_splitncnn_1
Convolution Conv_16 1 1 168_splitncnn_1 169 0=64 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=8192
ReLU LeakyRelu_17 1 1 169 170 0=1.000000e-01
Convolution Conv_18 1 1 170 171 0=128 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=73728
ReLU LeakyRelu_19 1 1 171 172 0=1.000000e-01
BinaryOp Add_20 2 1 172 168_splitncnn_0 173 0=0
Convolution Conv_21 1 1 173 174 0=256 1=3 11=3 2=1 12=1 3=2 13=2 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_22 1 1 174 175 0=1.000000e-01
Split splitncnn_3 1 2 175 175_splitncnn_0 175_splitncnn_1
Convolution Conv_23 1 1 175_splitncnn_1 176 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_24 1 1 176 177 0=1.000000e-01
Convolution Conv_25 1 1 177 178 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_26 1 1 178 179 0=1.000000e-01
BinaryOp Add_27 2 1 179 175_splitncnn_0 180 0=0
Split splitncnn_4 1 2 180 180_splitncnn_0 180_splitncnn_1
Convolution Conv_28 1 1 180_splitncnn_1 181 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_29 1 1 181 182 0=1.000000e-01
Convolution Conv_30 1 1 182 183 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_31 1 1 183 184 0=1.000000e-01
BinaryOp Add_32 2 1 184 180_splitncnn_0 185 0=0
Split splitncnn_5 1 2 185 185_splitncnn_0 185_splitncnn_1
Convolution Conv_33 1 1 185_splitncnn_1 186 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_34 1 1 186 187 0=1.000000e-01
Convolution Conv_35 1 1 187 188 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_36 1 1 188 189 0=1.000000e-01
BinaryOp Add_37 2 1 189 185_splitncnn_0 190 0=0
Split splitncnn_6 1 2 190 190_splitncnn_0 190_splitncnn_1
Convolution Conv_38 1 1 190_splitncnn_1 191 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_39 1 1 191 192 0=1.000000e-01
Convolution Conv_40 1 1 192 193 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_41 1 1 193 194 0=1.000000e-01
BinaryOp Add_42 2 1 194 190_splitncnn_0 195 0=0
Split splitncnn_7 1 2 195 195_splitncnn_0 195_splitncnn_1
Convolution Conv_43 1 1 195_splitncnn_1 196 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_44 1 1 196 197 0=1.000000e-01
Convolution Conv_45 1 1 197 198 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_46 1 1 198 199 0=1.000000e-01
BinaryOp Add_47 2 1 199 195_splitncnn_0 200 0=0
Split splitncnn_8 1 2 200 200_splitncnn_0 200_splitncnn_1
Convolution Conv_48 1 1 200_splitncnn_1 201 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_49 1 1 201 202 0=1.000000e-01
Convolution Conv_50 1 1 202 203 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_51 1 1 203 204 0=1.000000e-01
BinaryOp Add_52 2 1 204 200_splitncnn_0 205 0=0
Split splitncnn_9 1 2 205 205_splitncnn_0 205_splitncnn_1
Convolution Conv_53 1 1 205_splitncnn_1 206 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_54 1 1 206 207 0=1.000000e-01
Convolution Conv_55 1 1 207 208 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_56 1 1 208 209 0=1.000000e-01
BinaryOp Add_57 2 1 209 205_splitncnn_0 210 0=0
Split splitncnn_10 1 2 210 210_splitncnn_0 210_splitncnn_1
Convolution Conv_58 1 1 210_splitncnn_1 211 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_59 1 1 211 212 0=1.000000e-01
Convolution Conv_60 1 1 212 213 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_61 1 1 213 214 0=1.000000e-01
BinaryOp Add_62 2 1 214 210_splitncnn_0 215 0=0
Split splitncnn_11 1 2 215 215_splitncnn_0 215_splitncnn_1
Convolution Conv_63 1 1 215_splitncnn_1 216 0=512 1=3 11=3 2=1 12=1 3=2 13=2 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_64 1 1 216 217 0=1.000000e-01
Split splitncnn_12 1 2 217 217_splitncnn_0 217_splitncnn_1
Convolution Conv_65 1 1 217_splitncnn_1 218 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_66 1 1 218 219 0=1.000000e-01
Convolution Conv_67 1 1 219 220 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_68 1 1 220 221 0=1.000000e-01
BinaryOp Add_69 2 1 221 217_splitncnn_0 222 0=0
Split splitncnn_13 1 2 222 222_splitncnn_0 222_splitncnn_1
Convolution Conv_70 1 1 222_splitncnn_1 223 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_71 1 1 223 224 0=1.000000e-01
Convolution Conv_72 1 1 224 225 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_73 1 1 225 226 0=1.000000e-01
BinaryOp Add_74 2 1 226 222_splitncnn_0 227 0=0
Split splitncnn_14 1 2 227 227_splitncnn_0 227_splitncnn_1
Convolution Conv_75 1 1 227_splitncnn_1 228 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_76 1 1 228 229 0=1.000000e-01
Convolution Conv_77 1 1 229 230 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_78 1 1 230 231 0=1.000000e-01
BinaryOp Add_79 2 1 231 227_splitncnn_0 232 0=0
Split splitncnn_15 1 2 232 232_splitncnn_0 232_splitncnn_1
Convolution Conv_80 1 1 232_splitncnn_1 233 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_81 1 1 233 234 0=1.000000e-01
Convolution Conv_82 1 1 234 235 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_83 1 1 235 236 0=1.000000e-01
BinaryOp Add_84 2 1 236 232_splitncnn_0 237 0=0
Split splitncnn_16 1 2 237 237_splitncnn_0 237_splitncnn_1
Convolution Conv_85 1 1 237_splitncnn_1 238 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_86 1 1 238 239 0=1.000000e-01
Convolution Conv_87 1 1 239 240 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_88 1 1 240 241 0=1.000000e-01
BinaryOp Add_89 2 1 241 237_splitncnn_0 242 0=0
Split splitncnn_17 1 2 242 242_splitncnn_0 242_splitncnn_1
Convolution Conv_90 1 1 242_splitncnn_1 243 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_91 1 1 243 244 0=1.000000e-01
Convolution Conv_92 1 1 244 245 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_93 1 1 245 246 0=1.000000e-01
BinaryOp Add_94 2 1 246 242_splitncnn_0 247 0=0
Split splitncnn_18 1 2 247 247_splitncnn_0 247_splitncnn_1
Convolution Conv_95 1 1 247_splitncnn_1 248 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_96 1 1 248 249 0=1.000000e-01
Convolution Conv_97 1 1 249 250 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_98 1 1 250 251 0=1.000000e-01
BinaryOp Add_99 2 1 251 247_splitncnn_0 252 0=0
Split splitncnn_19 1 2 252 252_splitncnn_0 252_splitncnn_1
Convolution Conv_100 1 1 252_splitncnn_1 253 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_101 1 1 253 254 0=1.000000e-01
Convolution Conv_102 1 1 254 255 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_103 1 1 255 256 0=1.000000e-01
BinaryOp Add_104 2 1 256 252_splitncnn_0 257 0=0
Split splitncnn_20 1 2 257 257_splitncnn_0 257_splitncnn_1
Convolution Conv_105 1 1 257_splitncnn_1 258 0=1024 1=3 11=3 2=1 12=1 3=2 13=2 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_106 1 1 258 259 0=1.000000e-01
Split splitncnn_21 1 2 259 259_splitncnn_0 259_splitncnn_1
Convolution Conv_107 1 1 259_splitncnn_1 260 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=524288
ReLU LeakyRelu_108 1 1 260 261 0=1.000000e-01
Convolution Conv_109 1 1 261 262 0=1024 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_110 1 1 262 263 0=1.000000e-01
BinaryOp Add_111 2 1 263 259_splitncnn_0 264 0=0
Split splitncnn_22 1 2 264 264_splitncnn_0 264_splitncnn_1
Convolution Conv_112 1 1 264_splitncnn_1 265 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=524288
ReLU LeakyRelu_113 1 1 265 266 0=1.000000e-01
Convolution Conv_114 1 1 266 267 0=1024 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_115 1 1 267 268 0=1.000000e-01
BinaryOp Add_116 2 1 268 264_splitncnn_0 269 0=0
Split splitncnn_23 1 2 269 269_splitncnn_0 269_splitncnn_1
Convolution Conv_117 1 1 269_splitncnn_1 270 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=524288
ReLU LeakyRelu_118 1 1 270 271 0=1.000000e-01
Convolution Conv_119 1 1 271 272 0=1024 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_120 1 1 272 273 0=1.000000e-01
BinaryOp Add_121 2 1 273 269_splitncnn_0 274 0=0
Split splitncnn_24 1 2 274 274_splitncnn_0 274_splitncnn_1
Convolution Conv_122 1 1 274_splitncnn_1 275 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=524288
ReLU LeakyRelu_123 1 1 275 276 0=1.000000e-01
Convolution Conv_124 1 1 276 277 0=1024 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_125 1 1 277 278 0=1.000000e-01
BinaryOp Add_126 2 1 278 274_splitncnn_0 279 0=0
Convolution Conv_127 1 1 279 280 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=524288
ReLU LeakyRelu_128 1 1 280 281 0=1.000000e-01
Convolution Conv_129 1 1 281 282 0=1024 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_130 1 1 282 283 0=1.000000e-01
Convolution Conv_131 1 1 283 284 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=524288
ReLU LeakyRelu_132 1 1 284 285 0=1.000000e-01
Split splitncnn_25 1 4 285 285_splitncnn_0 285_splitncnn_1 285_splitncnn_2 285_splitncnn_3
Pooling MaxPool_133 1 1 285_splitncnn_3 286 0=0 1=5 11=5 2=1 12=1 3=2 13=2 14=2 15=2 5=1
Pooling MaxPool_134 1 1 285_splitncnn_2 287 0=0 1=9 11=9 2=1 12=1 3=4 13=4 14=4 15=4 5=1
Pooling MaxPool_135 1 1 285_splitncnn_1 288 0=0 1=13 11=13 2=1 12=1 3=6 13=6 14=6 15=6 5=1
Concat Concat_136 4 1 288 287 286 285_splitncnn_0 289 0=0
Convolution Conv_137 1 1 289 290 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=1048576
ReLU LeakyRelu_138 1 1 290 291 0=1.000000e-01
Convolution Conv_139 1 1 291 292 0=1024 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_140 1 1 292 293 0=1.000000e-01
Convolution Conv_141 1 1 293 294 0=512 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=524288
ReLU LeakyRelu_142 1 1 294 295 0=1.000000e-01
Split splitncnn_26 1 2 295 295_splitncnn_0 295_splitncnn_1
Convolution Conv_143 1 1 295_splitncnn_1 296 0=1024 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=4718592
ReLU LeakyRelu_144 1 1 296 297 0=1.000000e-01
Convolution Conv_145 1 1 297 298 0=255 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=261120
Reshape Reshape_147 1 1 298 300 0=60 1=85 2=3
Permute Transpose_148 1 1 300 301 0=1
Reshape Reshape_150 1 1 301 303 0=85
Split splitncnn_27 1 4 303 303_splitncnn_0 303_splitncnn_1 303_splitncnn_2 303_splitncnn_3
Crop Slice_155 1 1 303_splitncnn_3 308 -23309=1,0 -23310=1,2 -23311=1,0
Sigmoid Sigmoid_156 1 1 308 309
BinaryOp Add_158 2 1 309 310 311 0=0
Crop Slice_163 1 1 303_splitncnn_2 316 -23309=1,2 -23310=1,4 -23311=1,0
UnaryOp Exp_164 1 1 316 317 0=7
BinaryOp Mul_166 2 1 317 318 319 0=2
Crop Slice_171 1 1 303_splitncnn_1 324 -23309=1,5 -23310=1,85 -23311=1,0
Sigmoid Sigmoid_172 1 1 324 325
Crop Slice_177 1 1 303_splitncnn_0 330 -23309=1,4 -23310=1,5 -23311=1,0
Sigmoid Sigmoid_178 1 1 330 331
BinaryOp Mul_179 2 1 325 331 332 0=2
BinaryOp Mul_181 2 1 311 333 334 0=2
Convolution Conv_182 1 1 295_splitncnn_0 335 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_183 1 1 335 336 0=1.000000e-01
Interp Resize_192 1 1 336 347 0=1 1=1.000000e+00 2=1.000000e+00 3=20 4=12 6=0
Concat Concat_193 2 1 347 257_splitncnn_0 348 0=0
Convolution Conv_194 1 1 348 349 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=196608
ReLU LeakyRelu_195 1 1 349 350 0=1.000000e-01
Convolution Conv_196 1 1 350 351 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_197 1 1 351 352 0=1.000000e-01
Convolution Conv_198 1 1 352 353 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_199 1 1 353 354 0=1.000000e-01
Convolution Conv_200 1 1 354 355 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_201 1 1 355 356 0=1.000000e-01
Convolution Conv_202 1 1 356 357 0=256 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=131072
ReLU LeakyRelu_203 1 1 357 358 0=1.000000e-01
Split splitncnn_28 1 2 358 358_splitncnn_0 358_splitncnn_1
Convolution Conv_204 1 1 358_splitncnn_1 359 0=512 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=1179648
ReLU LeakyRelu_205 1 1 359 360 0=1.000000e-01
Convolution Conv_206 1 1 360 361 0=255 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=130560
Reshape Reshape_208 1 1 361 363 0=240 1=85 2=3
Permute Transpose_209 1 1 363 364 0=1
Reshape Reshape_211 1 1 364 366 0=85
Split splitncnn_29 1 4 366 366_splitncnn_0 366_splitncnn_1 366_splitncnn_2 366_splitncnn_3
Crop Slice_216 1 1 366_splitncnn_3 371 -23309=1,0 -23310=1,2 -23311=1,0
Sigmoid Sigmoid_217 1 1 371 372
BinaryOp Add_219 2 1 372 373 374 0=0
Crop Slice_224 1 1 366_splitncnn_2 379 -23309=1,2 -23310=1,4 -23311=1,0
UnaryOp Exp_225 1 1 379 380 0=7
BinaryOp Mul_227 2 1 380 381 382 0=2
Crop Slice_232 1 1 366_splitncnn_1 387 -23309=1,5 -23310=1,85 -23311=1,0
Sigmoid Sigmoid_233 1 1 387 388
Crop Slice_238 1 1 366_splitncnn_0 393 -23309=1,4 -23310=1,5 -23311=1,0
Sigmoid Sigmoid_239 1 1 393 394
BinaryOp Mul_240 2 1 388 394 395 0=2
BinaryOp Mul_242 2 1 374 396 397 0=2
Convolution Conv_243 1 1 358_splitncnn_0 398 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_244 1 1 398 399 0=1.000000e-01
Interp Resize_253 1 1 399 410 0=1 1=1.000000e+00 2=1.000000e+00 3=40 4=24 6=0
Concat Concat_254 2 1 410 215_splitncnn_0 411 0=0
Convolution Conv_255 1 1 411 412 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=49152
ReLU LeakyRelu_256 1 1 412 413 0=1.000000e-01
Convolution Conv_257 1 1 413 414 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_258 1 1 414 415 0=1.000000e-01
Convolution Conv_259 1 1 415 416 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_260 1 1 416 417 0=1.000000e-01
Convolution Conv_261 1 1 417 418 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_262 1 1 418 419 0=1.000000e-01
Convolution Conv_263 1 1 419 420 0=128 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=32768
ReLU LeakyRelu_264 1 1 420 421 0=1.000000e-01
Convolution Conv_265 1 1 421 422 0=256 1=3 11=3 2=1 12=1 3=1 13=1 4=1 14=1 15=1 16=1 5=1 6=294912
ReLU LeakyRelu_266 1 1 422 423 0=1.000000e-01
Convolution Conv_267 1 1 423 424 0=255 1=1 11=1 2=1 12=1 3=1 13=1 4=0 14=0 15=0 16=0 5=1 6=65280
Reshape Reshape_269 1 1 424 426 0=960 1=85 2=3
Permute Transpose_270 1 1 426 427 0=1
Reshape Reshape_272 1 1 427 429 0=85
Split splitncnn_30 1 4 429 429_splitncnn_0 429_splitncnn_1 429_splitncnn_2 429_splitncnn_3
Crop Slice_277 1 1 429_splitncnn_3 434 -23309=1,0 -23310=1,2 -23311=1,0
Sigmoid Sigmoid_278 1 1 434 435
BinaryOp Add_280 2 1 435 436 437 0=0
Crop Slice_285 1 1 429_splitncnn_2 442 -23309=1,2 -23310=1,4 -23311=1,0
UnaryOp Exp_286 1 1 442 443 0=7
BinaryOp Mul_288 2 1 443 444 445 0=2
Crop Slice_293 1 1 429_splitncnn_1 450 -23309=1,5 -23310=1,85 -23311=1,0
Sigmoid Sigmoid_294 1 1 450 451
Crop Slice_299 1 1 429_splitncnn_0 456 -23309=1,4 -23310=1,5 -23311=1,0
Sigmoid Sigmoid_300 1 1 456 457
BinaryOp Mul_301 2 1 451 457 458 0=2
BinaryOp Mul_303 2 1 437 459 460 0=2
Concat Concat_304 3 1 332 395 458 classes 0=-1
Concat Concat_305 3 1 334 397 460 462 0=-1
Concat Concat_306 3 1 319 382 445 463 0=-1
Concat Concat_307 2 1 462 463 boxes 0=0
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
- 11
- 12
- 13
- 14
- 15
- 16
- 17
- 18
- 19
- 20
- 21
- 22
- 23
- 24
- 25
- 26
- 27
- 28
- 29
- 30
- 31
- 32
- 33
- 34
- 35
- 36
- 37
- 38
- 39
- 40
- 41
- 42
- 43
- 44
- 45
- 46
- 47
- 48
- 49
- 50
- 51
- 52
- 53
- 54
- 55
- 56
- 57
- 58
- 59
- 60
- 61
- 62
- 63
- 64
- 65
- 66
- 67
- 68
- 69
- 70
- 71
- 72
- 73
- 74
- 75
- 76
- 77
- 78
- 79
- 80
- 81
- 82
- 83
- 84
- 85
- 86
- 87
- 88
- 89
- 90
- 91
- 92
- 93
- 94
- 95
- 96
- 97
- 98
- 99
- 100
- 101
- 102
- 103
- 104
- 105
- 106
- 107
- 108
- 109
- 110
- 111
- 112
- 113
- 114
- 115
- 116
- 117
- 118
- 119
- 120
- 121
- 122
- 123
- 124
- 125
- 126
- 127
- 128
- 129
- 130
- 131
- 132
- 133
- 134
- 135
- 136
- 137
- 138
- 139
- 140
- 141
- 142
- 143
- 144
- 145
- 146
- 147
- 148
- 149
- 150
- 151
- 152
- 153
- 154
- 155
- 156
- 157
- 158
- 159
- 160
- 161
- 162
- 163
- 164
- 165
- 166
- 167
- 168
- 169
- 170
- 171
- 172
- 173
- 174
- 175
- 176
- 177
- 178
- 179
- 180
- 181
- 182
- 183
- 184
- 185
- 186
- 187
- 188
- 189
- 190
- 191
- 192
- 193
- 194
- 195
- 196
- 197
- 198
- 199
- 200
- 201
- 202
- 203
- 204
- 205
- 206
- 207
- 208
- 209
- 210
- 211
- 212
- 213
- 214
- 215
- 216
- 217
- 218
- 219
- 220
- 221
- 222
- 223
- 224
- 225
- 226
- 227
- 228
- 229
- 230
- 231
- 232
- 233
- 234
- 235
- 236
- 237
- 238
- 239
- 240
- 241
- 242
- 243
- 244
- 245
- 246
- 247
- 248
- 249
- 250
- 251
- 252
- 253
- 254
- 255
- 256
- 257
- 258
- 259
- 260
- 261
- 262
- 263
- 264
- 265
- 266
- 267
- 268
- 269
- 270
- 271
- 272
- 273
文章来源: positive.blog.csdn.net，作者：墨理学AI，版权归原作者所有，如需转载，请联系作者。
原文链接：positive.blog.csdn.net/article/details/120455373
【版权声明】本文为华为云社区用户转载文章，如果您发现本社区中有涉嫌抄袭的内容，欢迎发送邮件进行举报，并提供相关证据，一经查实，本社区将立刻删除涉嫌侵权内容，举报邮箱：
cloudbbs@huaweicloud.com
- 点赞
- 收藏
- 关注作者
评论（0）