Skip to content

Commit 52abfa8

Browse files
authored
Merge pull request #63 from JDAI-CV/update
Check Android SDK in runtime, support sigmoid and misc changes
2 parents 7e70872 + c969332 commit 52abfa8

File tree

10 files changed

+517
-353
lines changed

10 files changed

+517
-353
lines changed

common/daq.fbs

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ enum DataType:byte { Float32 = 0, Int8, Int32, Float16, Bool8,
77
enum FuseCode:byte { None = 0, Relu, Relu1, Relu6 }
88
enum LayerType:byte { Conv2D = 0, AvePool, MaxPool, Relu, Softmax, FC, Add, Concat,
99
DepthwiseConv2D, BatchToSpace, SpaceToBatch, StridedSlice, Mul, AddScalar, MulScalar,
10-
Dequantize, LRN, Tanh, Floor}
10+
Dequantize, LRN, Tanh, Floor, Logistic}
1111

1212
table Tensor {
1313
data_type:DataType;
@@ -174,6 +174,11 @@ table Floor {
174174
output:string;
175175
}
176176

177+
table Logistic {
178+
input:string;
179+
output:string;
180+
}
181+
177182
table Layer {
178183
type:LayerType;
179184
conv2d_param:Conv2D;
@@ -195,6 +200,7 @@ table Layer {
195200
lrn_param:LRN;
196201
tanh_param:Tanh;
197202
floor_param:Floor;
203+
logistic_param:Logistic;
198204
}
199205

200206
table Model {

dnnlibrary/DaqReader.cpp

Lines changed: 31 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -19,44 +19,48 @@ void ReadDaqImpl(const uint8_t *buf, ModelBuilder &builder);
1919

2020
std::string layer_type_to_str(DNN::LayerType type) {
2121
switch (type) {
22+
// DaqReader auto generated layer_type_to_str start
23+
case DNN::LayerType::Conv2D:
24+
return "Conv2D";
25+
case DNN::LayerType::AvePool:
26+
return "AvePool";
27+
case DNN::LayerType::MaxPool:
28+
return "MaxPool";
29+
case DNN::LayerType::Relu:
30+
return "Relu";
31+
case DNN::LayerType::Softmax:
32+
return "Softmax";
2233
case DNN::LayerType::FC:
23-
return "fc";
34+
return "FC";
2435
case DNN::LayerType::Add:
2536
return "Add";
26-
case DNN::LayerType::Relu:
27-
return "relu";
28-
case DNN::LayerType::Conv2D:
29-
return "conv";
3037
case DNN::LayerType::Concat:
31-
return "concat";
32-
case DNN::LayerType::MaxPool:
33-
return "maxpool";
34-
case DNN::LayerType::AvePool:
35-
return "avepool";
36-
case DNN::LayerType::Softmax:
37-
return "softmax";
38+
return "Concat";
3839
case DNN::LayerType::DepthwiseConv2D:
39-
return "depthwsie";
40+
return "DepthwiseConv2D";
4041
case DNN::LayerType::BatchToSpace:
41-
return "batch2space";
42+
return "BatchToSpace";
4243
case DNN::LayerType::SpaceToBatch:
43-
return "space2batch";
44+
return "SpaceToBatch";
4445
case DNN::LayerType::StridedSlice:
45-
return "stridedslice";
46+
return "StridedSlice";
4647
case DNN::LayerType::Mul:
47-
return "mul";
48-
case DNN::LayerType::MulScalar:
49-
return "mulscalar";
48+
return "Mul";
5049
case DNN::LayerType::AddScalar:
51-
return "addscalar";
50+
return "AddScalar";
51+
case DNN::LayerType::MulScalar:
52+
return "MulScalar";
5253
case DNN::LayerType::Dequantize:
53-
return "dequantize";
54+
return "Dequantize";
5455
case DNN::LayerType::LRN:
5556
return "LRN";
5657
case DNN::LayerType::Tanh:
57-
return "tanh";
58+
return "Tanh";
5859
case DNN::LayerType::Floor:
59-
return "floor";
60+
return "Floor";
61+
case DNN::LayerType::Logistic:
62+
return "Logistic";
63+
// DaqReader auto generated layer_type_to_str end
6064
}
6165
}
6266

@@ -275,29 +279,19 @@ void AddLayers(const DNN::Model &model, ModelBuilder &builder) {
275279
break;
276280
}
277281
case DNN::LayerType::BatchToSpace: {
278-
#if __ANDROID_API__ >= __ANDROID_API_P__
279282
ADD_LAYER(batch_to_space, BatchToSpaceND, input, block_sizes,
280283
output);
281284
break;
282-
#endif
283285
}
284286
case DNN::LayerType::SpaceToBatch: {
285-
#if __ANDROID_API__ >= __ANDROID_API_P__
286287
ADD_LAYER(space_to_batch, SpaceToBatchND, input, block_sizes,
287288
pads, output);
288289
break;
289-
#endif
290290
}
291291
case DNN::LayerType::StridedSlice: {
292-
#if __ANDROID_API__ >= __ANDROID_API_P__
293292
ADD_LAYER(strided_slice, StridedSlice, input, starts, ends,
294293
strides, begin_mask, end_mask, shrink_axis_mask,
295294
output);
296-
#else
297-
throw std::invalid_argument("Unsupported layer " +
298-
layer_type_to_str(layer->type()) +
299-
" in API 28");
300-
#endif
301295
break;
302296
}
303297
case DNN::LayerType::LRN: {
@@ -312,6 +306,10 @@ void AddLayers(const DNN::Model &model, ModelBuilder &builder) {
312306
ADD_LAYER(floor, Floor, input, output);
313307
break;
314308
}
309+
case DNN::LayerType::Logistic: {
310+
ADD_LAYER(logistic, Logistic, input, output);
311+
break;
312+
}
315313
}
316314
}
317315
}

0 commit comments

Comments
 (0)