零基础导入自己YOLOv3或YOLOv3-Tiny模型

goenjoy · April 19, 2022, 3:43am

@Frank 麻烦回复下，谢谢！！！！！！

Frank · April 19, 2022, 12:13pm

@purplepetal 置信度在你的模型训练过程中去设置就可以了。这里的转换应用，不需要设置置信度

purplepetal · April 21, 2022, 7:11am

@Frank 我使用的pytorch版的yolov3（ github.com），在train时并没有置信度，在test和detect里才有，而且我运行这个模型也就是detect，是可以调整它的置信度呀

Frank · April 21, 2022, 7:58am

@purplepetal 以我们的yolov3 demo为例，配置位置在这里

github.com

khadas/aml_npu_app/blob/master/DDK_6.3.3.4/detect_library/model_code/detect_yolo_v3/yolov3_process.c#L304

      
        
            nn_height = tensor->attr.size[1];

            nn_channel = tensor->attr.size[2];

            (void)nn_channel;

            int size[3]={nn_width/32, nn_height/32,85*3};

            

            int sz[10];

            int i, j, stride;

            int output_cnt = 0;

            int output_len = 0;

            int num_class = 80;

            float threshold = 0.5;

            float iou_threshold = 0.4;

            uint8_t *tensor_data = NULL;

            float *predictions = NULL;

            

            for (i = 0; i < graph->output.num; i++) {

                tensor = vsi_nn_GetTensor(graph, graph->output.tensors[i]);

                sz[i] = 1;

                for (j = 0; j < tensor->attr.dim_num; j++) {

                    sz[i] *= tensor->attr.size[j];

                }

你根据你自己的模型修改你的代码。

purplepetal · April 22, 2022, 3:43am

好的，非常感谢！！！！！！！！！！！！！！！！！！！！！！