search

karta0807913 / caffe-yolo-9000

caffe-yolo-9000
Other
48 stars 16 forks source link

The concat and reorg layer error #8

Open VisionZQ opened 6 years ago

VisionZQ commented 6 years ago

I have trained the model as instruction. Then, I use NetScope to draw the network structure diagram,but it's poor error in the reorg layer, the diagram as shown below. 2 After I learn the darknet yolo v2 , I think that the new diagram is more more reasonable. 1

zhengstake commented 6 years ago

I think this is the artifact of in-place operation of the second reshape layer. If you change the layer name of the second reshape layer to 'reshape2', for example, the graph flow would be correct, right?

DarkNet_graph.pdf DarkNet_graph2.pdf