Skip to content

模型推理会很耗时 #10

@zhongchaoze

Description

@zhongchaoze

for(int i = 0; i < grid_h; i++){//row
for(int j = 0; j < grid_w; j++){//col
//int8_t i8_max_score = -255;
int8_t i8_max_score = -128;
int cur_label = 0;
for(int a = 0; a < OBJ_CLASS_NUM; a++){
int32_t class_index = a*grid_len + i * grid_w + j;
//printf("%d ", class_index);
int8_t cls_confidence = input[class_index];
if(cls_confidence > i8_max_score){
i8_max_score = cls_confidence;
cur_label = a;
}
}
}
每一个采样率下都得遍历一次类别,导致后处理时间特别长!!!!!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions