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

推理结束的层的索引,一般不填