10. 附录 1¶
10.1. 支持的layer¶
A
active arg AnnotatedData Accuracy
B
Bias Batchnorm BatchToSpace BroadcastBinary BN
C
Compare Concat ConstBinary Convolution Crop CPU
D
Data Deconvolution DetectionOutput Dropout DummyData
E
ELU Eltwise EltwiseBinary ExpandDims
F
Flatten
I
InnerProduct Interp ImageData
L
LRN
N
Normalize
O
Output
P
Prelu PSROIPooling Pad Permute Pooling PoolingTF PriorBox
R
ROIPooling RPN Relu Reciprocal Reduce ReduceFull Reorg Reshape
S
Scale Select ShapAssign ShapeConst ShapeOp ShapePack ShapeRef ShapeSlice ShuffelChannel Sigmoid Slice Softmax SoftmaxWithLoss SpaceToBatch Split SplitTF squeeze StrideSlice
T
Tile TopK Transpose
U
Upsample upsampleCopy
Y
Yolov3DetectionOutputupsampleCopy
10.2. U-FrameWork python接口¶
ufw.set_mode_cpu()
功能:设置网络工作在fp32 cpu模式下
输入参数:
Parameter |
Type |
Description |
无 |
ufw.set_mode_cpu_int8()
功能:设置网络工作在int8 cpu模式下
输入参数:
Parameter |
Type |
Description |
无 |
ufw.Net(model, weight)
功能:采用model,weight建立网络
输入参数:
Parameter |
Type |
Description |
model |
string |
表示网络结构的prototxt文件名 |
weight |
string |
表示网络系数的文件名 |
net.fill_blob_data({blob_name: input_data})
功能:
输入参数:
Parameter |
Type |
Description |
input |
dict |
输入blob名字和数据组成的字典 |
net.get_blob_data(blob_name)
功能:
输入参数:
Parameter |
Type |
Description |
blob_name |
string |
欲获取数据的blob的名字 |
net. forward (blobs=None, start=None, end=None, **kwargs)
功能: 进行网络推理,一般参数为空,进行整个网络推理
输入参数:
Parameter |
Type |
Description |
blobs |
string |
网络输出之外希望输出的blob的名字 |
start |
int |
推理开始的层的索引,一般不填 |
end |
int |
推理结束的层的索引,一般不填 |