可以用cnmon命令看一下。主要还是按照我发的.py文件做参考,试一下吧。是再官方平台上运行的
可以用cnmon命令看一下。主要还是按照我发的.py文件做参考,试一下吧。是再官方平台上运行的
dev-env-ubuntu: 构建基于Docker容器的寒武纪开发环境 - Gitee.com
我找到这个地址,下载到官方平台试了一下,还是相同的问题,
ct.set_core_number(1) 结果和ct.set_core_number(4) 结果相差很大
看着错误是input shape有问题。你查一下代码是否有问题。trace_input = torch.randn(1, 3, 640, 640, dtype=torch.float)[图片]展开
两个输出如下:
image 1/5 /home/yolov5/yolov5-4.0/data/images/1.jpg: num_boxes_final: 80.0
im0s.shape: (640, 640, 3)
[array([[ 530.5, 166.38, 550, 192.12, 0.33545, 0],
[ 149.12, 485.75, 208.38, 633.5, 0.27954, 0],
[ 150, 483, 207.62, 636.5, 0.84863, 1],
[ 284.75, 240.88, 310, 312, 0.79688, 1],
[ 234.38, 281.5, 258.75, 331.25, 0.78516, 1],
[ 488.25, 480.75, 545, 629, 0.78467, 1],
[ 262.75, 444.5, 313, 638, 0.77783, 1],
[ 97.812, 479.5, 135.75, 597, 0.72998, 1],
[ 447, 418.75, 486.75, 527.5, 0.69287, 1],
[ 307.5, 286, 330, 332, 0.67285, 1],
[ 465.5, 279, 493.75, 320.5, 0.58447, 1],
[ 504.25, 548.5, 571.5, 639, 0.36597, 1],
[ 80.438, 598, 144.38, 639, 0.88916, 2],
[ 119.75, 395, 148.62, 437.75, 0.84375, 2],
[ 226, 327.5, 258.75, 381.5, 0.8418, 2],
[ 174.38, 310.25, 205.62, 346.5, 0.83545, 2],
[ 71.375, 380, 103.5, 438.75, 0.81885, 2],
[ 129.25, 297.5, 161.38, 346.75, 0.74902, 2],
[ 168.5, 273.75, 217.62, 317.75, 0.74512, 2],
[ 171.62, 355.5, 207.12, 413, 0.72461, 2],
[ 97.938, 405.75, 121.38, 442.75, 0.66211, 2],
[ 158.75, 295.75, 181.62, 343, 0.6167, 2],
[ 145.12, 361.75, 176.25, 415, 0.53467, 2],
[ 107.56, 287.25, 132.25, 325.25, 0.50977, 2],
[ 100.38, 401.75, 145.38, 441.75, 0.26001, 2],
[ 79.562, 376.75, 94.688, 392, 0.4895, 3],
[ 486.5, 375, 525, 409, 0.87305, 4],
[ 378, 205.75, 398.75, 228.88, 0.8291, 4],
[ 450.25, 375.75, 489, 409.5, 0.80859, 4],
[ 507, 333.75, 540.5, 363, 0.75439, 4],
[ 498.75, 311.25, 531, 336.75, 0.75146, 4],
[ 568, 273.75, 593, 299, 0.74854, 4],
[ 447, 355.75, 481.75, 376.5, 0.72607, 4],
[ 213.5, 203.12, 230.88, 225.62, 0.72607, 4],
[ 192.62, 209.62, 213.75, 234.12, 0.71875, 4],
[ 478.75, 356.25, 516, 378, 0.70215, 4],
[ 326.75, 202.25, 349.5, 228.88, 0.69238, 4],
[ 434, 222.75, 468.75, 258.25, 0.68506, 4],
[ 357, 206, 374, 227.88, 0.67627, 4],
[ 465, 329.25, 502, 355.25, 0.67285, 4],
[ 499.25, 290, 529.5, 315.25, 0.66943, 4],
[ 555, 346.25, 576.5, 364.5, 0.66846, 4],
[ 237.12, 192.5, 259.5, 217.5, 0.66113, 4],
[ 618, 394.25, 639.5, 435.75, 0.65479, 4],
[ 485.75, 222.25, 513, 249.38, 0.63965, 4],
[ 623, 361.25, 639, 394.25, 0.61035, 4],
[ 503, 267.75, 531.5, 295.5, 0.60059, 4],
[ 624, 304, 639.5, 331.25, 0.59033, 4],
[ 532, 272.5, 560.5, 303.75, 0.5752, 4],
[ 589, 304.5, 614.5, 331.25, 0.55029, 4],
[ 616.5, 449.25, 640, 487, 0.54639, 4],
[ 591.5, 325.75, 616.5, 348, 0.54297, 4],
[ 570, 321.75, 596.5, 342.75, 0.54102, 4],
[ 573.5, 440, 609, 492.75, 0.52734, 4],
[ 529, 166.62, 550.5, 190.88, 0.52441, 4],
[ 576.5, 340.75, 601.5, 365, 0.521, 4],
[ 614.5, 480, 640, 516.5, 0.50781, 4],
[ 547.5, 300.75, 573.5, 326.75, 0.50244, 4],
[ 588.5, 276.25, 620, 305.25, 0.49854, 4],
[ 625, 328.25, 639.5, 360.5, 0.49854, 4],
[ 591.5, 341.75, 613, 363, 0.48267, 4],
[ 554, 324.25, 577, 344, 0.47925, 4],
[ 570, 301.25, 601, 327.5, 0.47412, 4],
[ 551, 234.62, 583.5, 262.75, 0.46802, 4],
[ 338.75, 204.62, 359.5, 229.38, 0.4519, 4],
[ 582.5, 504.25, 637, 551.5, 0.43408, 4],
[ 469.5, 344.5, 511.25, 365.25, 0.43335, 4],
[ 460.75, 322, 498.75, 345.25, 0.40771, 4],
[ 169.62, 211.38, 184.5, 233.75, 0.39966, 4],
[ 607, 274.5, 639, 307.5, 0.37842, 4],
[ 616, 503, 640, 540, 0.32959, 4],
[ 502.75, 313.75, 536.5, 358, 0.30933, 4],
[ 544.5, 327.5, 568, 345, 0.30737, 4],
[ 363.25, 208.88, 382, 228.75, 0.30005, 4],
[ 625.5, 337.5, 640, 376.75, 0.29272, 4],
[ 448.75, 358.75, 484, 396.5, 0.29224, 4],
[ 445.25, 292, 469, 319.75, 0.28516, 4],
[ 535, 167.25, 565, 190.25, 0.28101, 4],
[ 556.5, 324.5, 578, 364.5, 0.27173, 4],
[ 434.5, 251.38, 466.5, 274.5, 0.27148, 4]])]
错误的:
batchNum: 1
image 1/5 /home/yolov5/yolov5-4.0/data/images/1.jpg: num_boxes_final: 82.0
im0s.shape: (640, 640, 3)
[array([[ -127.38, -152.88, -99.125, -123.25, -0.151, -0.085205],
[ 254.12, 493.25, 656, 837.5, 0.12659, 0.31274],
[ 231.62, 301.25, 301.5, 353.25, 0.55908, 0.15796],
[ -55.938, -64.562, 218.75, 186.38, 0.079468, 0.15271],
[ -122.69, -167.62, -51.938, -127.88, -0.2627, -0.075073],
[ -141.25, -113.19, 30.25, 8.9297, 0.32861, 0.25366],
[ -25.266, 62.906, 149.38, 375.5, 0.044037, 0.33057],
[ 464, -35, 1078, 138, 0.26685, 0.25171],
[ -127.88, -135.62, 564, 220.12, -0.17676, -0.13708],
[ 152.5, 27.156, 241.62, 61.594, -0.22632, 0.26147],
[ -167.25, -152.62, 73.75, -28.219, -0.12219, -0.18665],
[ -143.62, -19.484, 260.5, 41.156, 0.21436, -0.039612],
[ -168.5, -163.75, -94.375, 486.5, -0.25488, -0.22327],
[ 96.875, 153.75, 165.12, 269.5, 0.35669, 0.0215],
[ 59.438, 5.7148, 89.188, 18.734, -0.10498, 0.07843],
[ -116, -150, -71.312, 311.75, -0.080444, -0.13025],
[ -176.38, -141.38, -106.12, 64.125, -0.26709, -0.27539],
[ 160.12, -133.25, 280.25, 582.5, 0.16052, 0.77246],
[ 112.19, -164.38, 339.5, -129.12, -0.053253, 0.17578],
[ -118.5, -23.719, 21.422, 496.75, 0.00030518, -0.13818],
[ -71.312, 87.25, 91.625, 164.38, -0.15332, 0.10602],
[ -150.38, -113.12, 223.12, 7.6562, -0.18176, -0.25122],
[ -109.56, -10.203, 144.25, 35.188, -0.17114, 0.21558]])]
请登录后评论