notebooks/assets/yolov5_graph_visualize.svg

<?xml version="1.0" encoding="UTF-8" standalone="no"?>
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN"
 "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
<!-- Generated by graphviz version 2.49.1 (20210923.0004)
 -->
<!-- Pages: 1 -->
<svg width="919pt" height="865pt"
 viewBox="0.00 0.00 919.29 865.00" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">
<g id="graph0" class="graph" transform="scale(1 1) rotate(0) translate(4 861)">
<polygon fill="white" stroke="transparent" points="-4,4 -4,-861 915.29,-861 915.29,4 -4,4"/>
<text text-anchor="middle" x="455.65" y="-841.8" font-family="Times,serif" font-size="14.00">YOLOv5</text>
<g id="clust1" class="cluster">
<title>cluster_.73</title>
<polygon fill="none" stroke="black" points="8,-136 8,-718 886,-718 886,-136 8,-136"/>
<text text-anchor="middle" x="447" y="-702.8" font-family="Times,serif" font-size="14.00">model (YOLO)</text>
</g>
<g id="clust2" class="cluster">
<title>cluster_model..32</title>
<polygon fill="none" stroke="black" points="16,-280 16,-571 606,-571 606,-280 16,-280"/>
<text text-anchor="middle" x="311" y="-555.8" font-family="Times,serif" font-size="14.00">model.head (YOLOHead)</text>
</g>
<!-- input_x.1 -->
<g id="node1" class="node">
<title>input_x.1</title>
<ellipse fill="none" stroke="black" cx="806" cy="-816" rx="45.49" ry="18"/>
<text text-anchor="middle" x="806" y="-812.3" font-family="Times,serif" font-size="14.00">input_x.1</text>
</g>
<!-- .72 -->
<g id="node2" class="node">
<title>.72</title>
<polygon fill="none" stroke="black" points="896.5,-762 715.5,-762 715.5,-726 896.5,-726 896.5,-762"/>
<text text-anchor="middle" x="806" y="-740.3" font-family="Times,serif" font-size="14.00">transform (YOLOTransform)</text>
</g>
<!-- input_x.1&#45;&gt;.72 -->
<g id="edge1" class="edge">
<title>input_x.1&#45;&gt;.72</title>
<path fill="none" stroke="black" d="M806,-797.7C806,-789.98 806,-780.71 806,-772.11"/>
<polygon fill="black" stroke="black" points="809.5,-772.1 806,-762.1 802.5,-772.1 809.5,-772.1"/>
</g>
<!-- model.input_1 -->
<g id="node3" class="node">
<title>model.input_1</title>
<ellipse fill="none" stroke="black" cx="607" cy="-669" rx="63.89" ry="18"/>
<text text-anchor="middle" x="607" y="-665.3" font-family="Times,serif" font-size="14.00">model.input_1</text>
</g>
<!-- .72&#45;&gt;model.input_1 -->
<g id="edge2" class="edge">
<title>.72&#45;&gt;model.input_1</title>
<path fill="none" stroke="black" d="M759.59,-725.98C727.75,-714.29 685.63,-698.84 653.72,-687.14"/>
<polygon fill="black" stroke="black" points="654.76,-683.79 644.16,-683.63 652.34,-690.36 654.76,-683.79"/>
</g>
<!-- input_output_0 -->
<g id="node17" class="node">
<title>input_output_0</title>
<path fill="none" stroke="black" d="M892,-108C892,-108 824,-108 824,-108 818,-108 812,-102 812,-96 812,-96 812,-84 812,-84 812,-78 818,-72 824,-72 824,-72 892,-72 892,-72 898,-72 904,-78 904,-84 904,-84 904,-96 904,-96 904,-102 898,-108 892,-108"/>
<text text-anchor="middle" x="858" y="-86.3" font-family="Times,serif" font-size="14.00">stack div mul</text>
</g>
<!-- .72&#45;&gt;input_output_0 -->
<g id="edge19" class="edge">
<title>.72&#45;&gt;input_output_0</title>
<path fill="none" stroke="black" d="M881.72,-725.73C884.74,-723.46 887.53,-720.89 890,-718 924.95,-677.09 906,-651.81 906,-598 906,-598 906,-598 906,-233 906,-189.31 906.64,-176.4 890,-136 887.25,-129.33 883.35,-122.69 879.22,-116.65"/>
<polygon fill="black" stroke="black" points="881.83,-114.28 873.1,-108.27 876.18,-118.41 881.83,-114.28"/>
</g>
<!-- model..31 -->
<g id="node4" class="node">
<title>model..31</title>
<polygon fill="none" stroke="black" points="624,-615 392,-615 392,-579 624,-579 624,-615"/>
<text text-anchor="middle" x="508" y="-593.3" font-family="Times,serif" font-size="14.00">model.backbone (BackboneWithPAN)</text>
</g>
<!-- model.input_1&#45;&gt;model..31 -->
<g id="edge3" class="edge">
<title>model.input_1&#45;&gt;model..31</title>
<path fill="none" stroke="black" d="M584.54,-652.12C571.47,-642.88 554.78,-631.07 540.32,-620.85"/>
<polygon fill="black" stroke="black" points="542.29,-617.96 532.11,-615.04 538.25,-623.68 542.29,-617.96"/>
</g>
<!-- model.head.input_1 -->
<g id="node5" class="node">
<title>model.head.input_1</title>
<ellipse fill="none" stroke="black" cx="508" cy="-522" rx="83.39" ry="18"/>
<text text-anchor="middle" x="508" y="-518.3" font-family="Times,serif" font-size="14.00">model.head.input_1</text>
</g>
<!-- model..31&#45;&gt;model.head.input_1 -->
<g id="edge4" class="edge">
<title>model..31&#45;&gt;model.head.input_1</title>
<path fill="none" stroke="black" d="M508,-578.7C508,-570.25 508,-559.87 508,-550.37"/>
<polygon fill="black" stroke="black" points="511.5,-550.18 508,-540.18 504.5,-550.18 511.5,-550.18"/>
</g>
<!-- model.head.input_2 -->
<g id="node6" class="node">
<title>model.head.input_2</title>
<ellipse fill="none" stroke="black" cx="318" cy="-522" rx="83.39" ry="18"/>
<text text-anchor="middle" x="318" y="-518.3" font-family="Times,serif" font-size="14.00">model.head.input_2</text>
</g>
<!-- model..31&#45;&gt;model.head.input_2 -->
<g id="edge5" class="edge">
<title>model..31&#45;&gt;model.head.input_2</title>
<path fill="none" stroke="black" d="M439.21,-578.86C431.32,-576.43 423.43,-573.8 416,-571 395.62,-563.32 373.78,-552.86 355.9,-543.66"/>
<polygon fill="black" stroke="black" points="357.43,-540.51 346.94,-538.99 354.19,-546.72 357.43,-540.51"/>
</g>
<!-- model.head.input_3 -->
<g id="node7" class="node">
<title>model.head.input_3</title>
<ellipse fill="none" stroke="black" cx="124" cy="-522" rx="83.39" ry="18"/>
<text text-anchor="middle" x="124" y="-518.3" font-family="Times,serif" font-size="14.00">model.head.input_3</text>
</g>
<!-- model..31&#45;&gt;model.head.input_3 -->
<g id="edge6" class="edge">
<title>model..31&#45;&gt;model.head.input_3</title>
<path fill="none" stroke="black" d="M391.83,-594.85C340.3,-592.01 279.38,-585.44 226,-571 203.27,-564.85 179.38,-553.88 160.43,-543.98"/>
<polygon fill="black" stroke="black" points="162.03,-540.86 151.56,-539.23 158.72,-547.03 162.03,-540.86"/>
</g>
<!-- model..33 -->
<g id="node13" class="node">
<title>model..33</title>
<polygon fill="none" stroke="black" points="878,-324 614,-324 614,-288 878,-288 878,-324"/>
<text text-anchor="middle" x="746" y="-302.3" font-family="Times,serif" font-size="14.00">model.anchor_generator (AnchorGenerator)</text>
</g>
<!-- model..31&#45;&gt;model..33 -->
<g id="edge14" class="edge">
<title>model..31&#45;&gt;model..33</title>
<path fill="none" stroke="black" d="M598.04,-578.83C602.26,-576.53 606.28,-573.94 610,-571 689.13,-508.6 726.71,-387.58 740.06,-333.83"/>
<polygon fill="black" stroke="black" points="743.48,-334.56 742.41,-324.02 736.68,-332.93 743.48,-334.56"/>
</g>
<!-- model.head..92 -->
<g id="node8" class="node">
<title>model.head..92</title>
<polygon fill="none" stroke="black" points="597.5,-468 418.5,-468 418.5,-432 597.5,-432 597.5,-468"/>
<text text-anchor="middle" x="508" y="-446.3" font-family="Times,serif" font-size="14.00">model.head.head.0 (Conv2d)</text>
</g>
<!-- model.head.input_1&#45;&gt;model.head..92 -->
<g id="edge7" class="edge">
<title>model.head.input_1&#45;&gt;model.head..92</title>
<path fill="none" stroke="black" d="M508,-503.7C508,-495.98 508,-486.71 508,-478.11"/>
<polygon fill="black" stroke="black" points="511.5,-478.1 508,-468.1 504.5,-478.1 511.5,-478.1"/>
</g>
<!-- model.head..93 -->
<g id="node9" class="node">
<title>model.head..93</title>
<polygon fill="none" stroke="black" points="400.5,-468 221.5,-468 221.5,-432 400.5,-432 400.5,-468"/>
<text text-anchor="middle" x="311" y="-446.3" font-family="Times,serif" font-size="14.00">model.head.head.1 (Conv2d)</text>
</g>
<!-- model.head.input_2&#45;&gt;model.head..93 -->
<g id="edge8" class="edge">
<title>model.head.input_2&#45;&gt;model.head..93</title>
<path fill="none" stroke="black" d="M316.27,-503.7C315.5,-495.98 314.57,-486.71 313.71,-478.11"/>
<polygon fill="black" stroke="black" points="317.19,-477.71 312.71,-468.1 310.22,-478.4 317.19,-477.71"/>
</g>
<!-- model.head..94 -->
<g id="node10" class="node">
<title>model.head..94</title>
<polygon fill="none" stroke="black" points="203.5,-468 24.5,-468 24.5,-432 203.5,-432 203.5,-468"/>
<text text-anchor="middle" x="114" y="-446.3" font-family="Times,serif" font-size="14.00">model.head.head.2 (Conv2d)</text>
</g>
<!-- model.head.input_3&#45;&gt;model.head..94 -->
<g id="edge9" class="edge">
<title>model.head.input_3&#45;&gt;model.head..94</title>
<path fill="none" stroke="black" d="M121.53,-503.7C120.43,-495.98 119.1,-486.71 117.87,-478.11"/>
<polygon fill="black" stroke="black" points="121.32,-477.51 116.44,-468.1 114.39,-478.5 121.32,-477.51"/>
</g>
<!-- input_model.head.output_0 -->
<g id="node12" class="node">
<title>input_model.head.output_0</title>
<path fill="none" stroke="black" d="M394.5,-396C394.5,-396 325.5,-396 325.5,-396 319.5,-396 313.5,-390 313.5,-384 313.5,-384 313.5,-372 313.5,-372 313.5,-366 319.5,-360 325.5,-360 325.5,-360 394.5,-360 394.5,-360 400.5,-360 406.5,-366 406.5,-372 406.5,-372 406.5,-384 406.5,-384 406.5,-390 400.5,-396 394.5,-396"/>
<text text-anchor="middle" x="360" y="-374.3" font-family="Times,serif" font-size="14.00">permute view</text>
</g>
<!-- model.head..92&#45;&gt;input_model.head.output_0 -->
<g id="edge12" class="edge">
<title>model.head..92&#45;&gt;input_model.head.output_0</title>
<path fill="none" stroke="black" d="M471.8,-431.88C451.74,-422.39 426.63,-410.51 405.29,-400.42"/>
<polygon fill="black" stroke="black" points="406.62,-397.18 396.09,-396.07 403.63,-403.51 406.62,-397.18"/>
</g>
<!-- model.head..93&#45;&gt;input_model.head.output_0 -->
<g id="edge11" class="edge">
<title>model.head..93&#45;&gt;input_model.head.output_0</title>
<path fill="none" stroke="black" d="M323.11,-431.7C328.87,-423.47 335.86,-413.48 342.21,-404.42"/>
<polygon fill="black" stroke="black" points="345.16,-406.3 348.03,-396.1 339.42,-402.29 345.16,-406.3"/>
</g>
<!-- model.head..94&#45;&gt;input_model.head.output_0 -->
<g id="edge10" class="edge">
<title>model.head..94&#45;&gt;input_model.head.output_0</title>
<path fill="none" stroke="black" d="M173.86,-431.97C213.29,-420.75 264.59,-406.15 303.35,-395.12"/>
<polygon fill="black" stroke="black" points="304.76,-398.36 313.42,-392.25 302.85,-391.62 304.76,-398.36"/>
</g>
<!-- model.head.output_0 -->
<g id="node11" class="node">
<title>model.head.output_0</title>
<ellipse fill="none" stroke="black" cx="410" cy="-306" rx="87.99" ry="18"/>
<text text-anchor="middle" x="410" y="-302.3" font-family="Times,serif" font-size="14.00">model.head.output_0</text>
</g>
<!-- model..34 -->
<g id="node14" class="node">
<title>model..34</title>
<polygon fill="none" stroke="black" points="849.5,-252 642.5,-252 642.5,-216 849.5,-216 849.5,-252"/>
<text text-anchor="middle" x="746" y="-230.3" font-family="Times,serif" font-size="14.00">model.post_process (PostProcess)</text>
</g>
<!-- model.head.output_0&#45;&gt;model..34 -->
<g id="edge15" class="edge">
<title>model.head.output_0&#45;&gt;model..34</title>
<path fill="none" stroke="black" d="M458.99,-290.88C472.84,-287.11 487.96,-283.2 502,-280 544.67,-270.28 591.91,-261.19 633.36,-253.76"/>
<polygon fill="black" stroke="black" points="633.99,-257.2 643.22,-252 632.76,-250.31 633.99,-257.2"/>
</g>
<!-- input_model.head.output_0&#45;&gt;model.head.output_0 -->
<g id="edge13" class="edge">
<title>input_model.head.output_0&#45;&gt;model.head.output_0</title>
<path fill="none" stroke="black" d="M372.36,-359.7C378.23,-351.47 385.37,-341.48 391.84,-332.42"/>
<polygon fill="black" stroke="black" points="394.82,-334.28 397.78,-324.1 389.12,-330.21 394.82,-334.28"/>
</g>
<!-- model..33&#45;&gt;model..34 -->
<g id="edge16" class="edge">
<title>model..33&#45;&gt;model..34</title>
<path fill="none" stroke="black" d="M746,-287.7C746,-279.98 746,-270.71 746,-262.11"/>
<polygon fill="black" stroke="black" points="749.5,-262.1 746,-252.1 742.5,-262.1 749.5,-262.1"/>
</g>
<!-- model.output_0 -->
<g id="node15" class="node">
<title>model.output_0</title>
<ellipse fill="none" stroke="black" cx="778" cy="-162" rx="68.49" ry="18"/>
<text text-anchor="middle" x="778" y="-158.3" font-family="Times,serif" font-size="14.00">model.output_0</text>
</g>
<!-- model..34&#45;&gt;model.output_0 -->
<g id="edge17" class="edge">
<title>model..34&#45;&gt;model.output_0</title>
<path fill="none" stroke="black" d="M753.91,-215.7C757.55,-207.73 761.95,-198.1 765.99,-189.26"/>
<polygon fill="black" stroke="black" points="769.21,-190.65 770.18,-180.1 762.84,-187.74 769.21,-190.65"/>
</g>
<!-- model.output_0&#45;&gt;input_output_0 -->
<g id="edge18" class="edge">
<title>model.output_0&#45;&gt;input_output_0</title>
<path fill="none" stroke="black" d="M796.96,-144.41C807,-135.63 819.51,-124.68 830.59,-114.99"/>
<polygon fill="black" stroke="black" points="833.12,-117.42 838.34,-108.2 828.51,-112.15 833.12,-117.42"/>
</g>
<!-- output_0 -->
<g id="node16" class="node">
<title>output_0</title>
<ellipse fill="none" stroke="black" cx="858" cy="-18" rx="42.79" ry="18"/>
<text text-anchor="middle" x="858" y="-14.3" font-family="Times,serif" font-size="14.00">output_0</text>
</g>
<!-- input_output_0&#45;&gt;output_0 -->
<g id="edge20" class="edge">
<title>input_output_0&#45;&gt;output_0</title>
<path fill="none" stroke="black" d="M858,-71.7C858,-63.98 858,-54.71 858,-46.11"/>
<polygon fill="black" stroke="black" points="861.5,-46.1 858,-36.1 854.5,-46.1 861.5,-46.1"/>
</g>
</g>
</svg>