Which system do you use? Android, Ubuntu, OOWOW or others?
Ubuntu
Which version of system do you use? Please provide the version of the system here:
Ubuntu 22.04
Please describe your issue below:
While converting YOLO, I saw the following values set when setting the input/output.
typedef enum {
AML_INPUT_DEFAULT = 0, //channel format: caffe 2 1 0 ,others 0 1 2
AML_INPUT_MODEL_NHWC = 1, //channel format: 0 1 2
AML_INPUT_MODEL_NCHW = 2, //channel format: 2 1 0
} aml_input_format_t;
typedef enum {
AML_OUTPUT_ORDER_DEFAULT = 0, //output format: nhwc for adla, nchw for galcore
AML_OUTPUT_ORDER_NHWC = 1, //output format: nhwc
AML_OUTPUT_ORDER_NCHW = 2, //output format: nchw
} aml_output_order_t;
Also, when I checked the input/output of the converted yolo model, it was in HWC structure like below.
Inputs : [0 ]
Input[0] : 0
Dim Count : 4
Size of dim[0] : 1
Size of dim[1] : 416
Size of dim[2] : 416
Size of dim[3] : 3
type : Fp32
inputs_memory_size : 2028.00 kb
Outputs : [1131 1150 1112 ]
Output[0] : 1131
Dim Count : 4
Size of dim[0] : 1
Size of dim[1] : 26
Size of dim[2] : 26
Size of dim[3] : 72
type : Fp32
Output[1] : 1150
Dim Count : 4
Size of dim[0] : 1
Size of dim[1] : 52
Size of dim[2] : 52
Size of dim[3] : 72
type : Fp32
Output[2] : 1112
Dim Count : 4
Size of dim[0] : 1
Size of dim[1] : 13
Size of dim[2] : 13
Size of dim[3] : 72
type : Fp32
outputs_memory_size : 998.16 kb
config:
max_outstanding_outputs : 4
axi_sram_size : 1277952 bytes