Home Neural Network Coaching Extensions: the best way to change the output layer to trust rating?

Coaching Extensions: the best way to change the output layer to trust rating?

0
Coaching Extensions: the best way to change the output layer to trust rating?

[ad_1]

Hello, I am engaged on OpenVINO coaching extensions to coach object detection mannequin with customized dataset.

 

I efficiently educated mannequin however I discovered the output layer solely contains packing containers and labels.

Je6_0-1709157130013.png

I want confidence rating to implement my software.

 

Earlier than making an attempt to make use of customized mannequin, I used ssdlite_mobilenet_v2. And this mannequin’s output is  DetectionOutput (every detection has the format: [image_id, label, conf, x_min, y_min, x_max, y_max]). 

I ponder why the default construction for object detection(in openvino coaching extensions) mannequin would not have a confidence rating.

 

I’ve looking associated paperwork and code however have not discovered reply..

How can I modify this construction to incorporate confidence rating?

 

Thanks.

[ad_2]