frame

Howdy, Stranger!

It looks like you're new here. If you want to get involved, click one of these buttons!

Sign In

Howdy, Stranger!

It looks like you're new here. If you want to get involved, click one of these buttons!

Tensorflow Yolo V2 model not working after being converted to NCS

Hello,

I'm having trouble with predictions using a custom-trained graph converted from tensorflow tiny-yolo-v2 to NCS graph.

I trained a 1-class classifier and I modified the run.py example and adapt it from 20 to 1 class. (Only modified all the 125 instances to 30 for filters and the value 20 to 1 for classes)

The example is running but it returns a very bizarre response.

I tried the custom-trained model before conversion and is working perfectly. It returns a very good result on predictions, the compiled graph is giving hundred of invalid boxes and not a single one corresponds to ground-truth.

Is there anything that has to be done before compiling? Why is my model not working on the graph. Why is the graph not working but the original .pb does.

I used this command to compile

mvNCCompile -o output/ncs_tiny_yolo_v2.graph built_graph/tiny-yolo-voc-1c.pb -s 12

Comments

This discussion has been closed.