How to use exported coreML model ?

[tommypham]

❔Question

How can I use output of model:

input {
  name: "image"
  type {
    imageType {
      width: 416
      height: 416
      colorSpace: RGB
    }
  }
}
output {
  name: "937"
  type {
    multiArrayType {
      dataType: FLOAT32
    }
  }
}
output {
  name: "950"
  type {
    multiArrayType {
      dataType: FLOAT32
    }
  }
}
output {
  name: "963"
  type {
    multiArrayType {
      dataType: FLOAT32
    }
  }
}
metadata {
  userDefined {
    key: "com.github.apple.coremltools.source"
    value: "torch==1.9.0"
  }
  userDefined {
    key: "com.github.apple.coremltools.version"
    value: "4.1"
  }
}

Additional context

The output I got is:

{'963': array([[[[[ 2.34705970e-01,  1.56080708e-01, -3.02680582e-01, ...,
           -5.84642649e+00, -5.60494471e+00, -5.13564110e+00],
          [-2.77504325e-02,  3.02708387e-01,  1.02229215e-01, ...,
           -5.38047886e+00, -5.25067806e+00, -5.00561047e+00],
          [-2.18960285e-01,  2.74042249e-01,  4.36938971e-01, ...,
           -5.33655214e+00, -5.07386541e+00, -4.90731621e+00],
          ...,
          [ 5.64223826e-02,  2.07174629e-01,  3.77338290e-01, ...,
           -5.71573734e+00, -4.99090195e+00, -4.45711613e+00],
          [ 2.44704992e-01,  4.28844929e-01, -1.45964593e-01, ...,
           -5.46977043e+00, -4.87067223e+00, -4.29896736e+00],
          [-4.12815511e-01,  2.11307526e-01, -5.54968059e-01, ...,
           -5.66117859e+00, -5.47401237e+00, -4.63303232e+00]],

         [[ 4.93963569e-01,  5.42731643e-01, -5.64413548e-01, ...,
           -5.79922104e+00, -5.68290377e+00, -5.34549618e+00],
          [ 1.12981245e-01,  2.81508923e-01, -8.77021477e-02, ...,
           -5.18312502e+00, -5.06729126e+00, -4.88984108e+00],
          [-5.66045761e-01,  2.18827933e-01,  2.67060667e-01, ...,
           -4.83452749e+00, -4.85995150e+00, -4.93088579e+00],
          ...,
          [ 2.39220530e-01, -4.80206430e-01,  3.30654263e-01, ...,
           -5.29895592e+00, -5.07235670e+00, -4.94823074e+00],
          [ 2.53607482e-01,  4.77914512e-03, -2.14182138e-01, ...,
           -4.98970175e+00, -4.76023674e+00, -4.56910324e+00],
          [-4.91524756e-01, -4.65967357e-02, -6.79589689e-01, ...,
           -5.42939091e+00, -5.46881771e+00, -4.77563667e+00]],

         [[ 4.76621568e-01, -2.38993257e-01, -6.11090302e-01, ...,
           -5.62920904e+00, -5.83080864e+00, -5.58471966e+00],
          [-9.96659845e-02, -1.72392458e-01, -1.55650407e-01, ...,
           -5.50657129e+00, -5.30944920e+00, -5.29805088e+00],
          [-6.30534887e-01, -1.57056063e-01,  2.16441497e-01, ...,
           -5.32671738e+00, -5.17464781e+00, -5.32916927e+00],
          ...,
          [ 4.35144365e-01, -3.36426467e-01,  2.62644738e-01, ...,
           -5.29718590e+00, -5.61095810e+00, -5.39254618e+00],
          [ 3.07465822e-01, -1.18429527e-01, -2.70846158e-01, ...,
           -5.36274719e+00, -5.19495535e+00, -5.16195297e+00],
          [-5.59656858e-01, -2.47284681e-01, -6.52646899e-01, ...,
           -5.58271027e+00, -5.70874691e+00, -5.15569687e+00]],

         ...,

         [[ 8.11232388e-01, -3.45519781e-02, -6.46291196e-01, ...,
           -5.59422827e+00, -6.20509434e+00, -5.75031376e+00],
          [-4.97394562e-01, -4.73909527e-02, -3.25089365e-01, ...,
           -5.30205917e+00, -6.07568502e+00, -5.67317963e+00],
          [-6.41372442e-01,  1.81404382e-01,  2.17511967e-01, ...,
           -5.49517012e+00, -6.33042574e+00, -5.92311668e+00],
          ...,
          [ 1.82669342e-01, -2.18181759e-01,  3.02736074e-01, ...,
           -5.32805729e+00, -5.79467773e+00, -5.41294146e+00],
          [ 5.61311603e-01, -6.65258616e-03, -3.58554155e-01, ...,
           -5.21306324e+00, -5.50296116e+00, -5.07404900e+00],
          [-4.67223525e-01, -1.12756938e-02, -7.70470917e-01, ...,
           -5.50148392e+00, -5.66896582e+00, -5.07891083e+00]],

         [[ 8.06507587e-01, -7.53435940e-02, -4.71196830e-01, ...,
           -5.62018681e+00, -6.11268187e+00, -5.49244118e+00],
          [-2.25397363e-01, -8.63676593e-02, -1.98898613e-01, ...,
           -5.47607422e+00, -5.88488197e+00, -5.33788681e+00],
          [-4.52402085e-01,  5.42433619e-01,  2.65873969e-01, ...,
           -5.75612736e+00, -6.12566900e+00, -5.70336485e+00],
          ...,
          [ 1.05671495e-01,  6.52270168e-02,  3.19947541e-01, ...,
           -5.29851341e+00, -5.49872875e+00, -4.98577785e+00],
          [ 1.98152795e-01,  7.22230971e-02, -2.36902058e-01, ...,
           -5.17087984e+00, -5.23863649e+00, -4.75454330e+00],
          [-5.33167243e-01, -2.32754350e-02, -6.92558587e-01, ...,
           -5.47804594e+00, -5.46276569e+00, -4.54630280e+00]],

         [[ 3.73654366e-01, -1.71311945e-01, -4.96528149e-01, ...,
           -5.81389427e+00, -6.11693907e+00, -5.60221863e+00],
          [ 9.67981070e-02, -4.97247040e-01, -1.40693590e-01, ...,
           -5.79419470e+00, -5.98088360e+00, -5.40378857e+00],
          [-2.64659762e-01, -2.69622833e-01,  3.18752706e-01, ...,
           -6.18358374e+00, -5.97611427e+00, -5.81422138e+00],
          ...,
          [ 5.14770746e-02, -4.75456297e-01,  3.15415025e-01, ...,
           -5.42153168e+00, -5.59720421e+00, -4.85159206e+00],
          [ 4.54064727e-01, -7.33817577e-01, -2.86898822e-01, ...,
           -5.28905821e+00, -5.35023260e+00, -4.49439049e+00],
          [-4.20190871e-01, -6.82255626e-01, -7.39860594e-01, ...,
           -5.73957586e+00, -5.55793333e+00, -4.64713192e+00]]],


        [[[ 4.03554380e-01,  6.97367907e-01, -5.04441977e-01, ...,
           -5.61771011e+00, -5.57402420e+00, -5.01351547e+00],
          [-1.10867269e-01,  1.13850689e+00,  1.35201663e-02, ...,
           -5.21486759e+00, -5.27738285e+00, -4.94242382e+00],
          [-5.36063433e-01,  1.07286406e+00,  9.78055000e-02, ...,
           -5.19615507e+00, -5.12057495e+00, -4.86221409e+00],
          ...,
          [ 4.59824353e-02,  9.43901598e-01,  1.25896662e-01, ...,
           -5.50127125e+00, -4.97301292e+00, -4.42343998e+00],
          [ 9.34889615e-02,  1.24610198e+00, -2.94172585e-01, ...,
           -5.24286604e+00, -4.86236668e+00, -4.27460480e+00],
          [-6.71207964e-01,  7.50388205e-01, -6.00107193e-01, ...,
           -5.34793329e+00, -5.34935284e+00, -4.50899792e+00]],

         [[ 7.10298240e-01,  7.57138014e-01, -7.57023156e-01, ...,
           -5.60092545e+00, -5.68016243e+00, -5.18378544e+00],
          [ 2.55081862e-01,  5.60603917e-01, -3.59498739e-01, ...,
           -5.04642630e+00, -5.17277050e+00, -4.84327412e+00],
          [-7.42206991e-01,  4.78325695e-01, -1.06343634e-01, ...,
           -4.76523685e+00, -5.04289722e+00, -4.91264725e+00],
          ...,
          [ 2.37569302e-01,  1.63025036e-01,  3.14418077e-02, ...,
           -5.14696121e+00, -5.14793110e+00, -4.91633606e+00],
          [ 5.23767397e-02,  4.30219471e-01, -4.33527678e-01, ...,
           -4.78927708e+00, -4.85385704e+00, -4.54001236e+00],
          [-9.39502954e-01,  2.99670339e-01, -7.45655417e-01, ...,
           -5.15938902e+00, -5.40793705e+00, -4.66286278e+00]],

         [[ 7.33329892e-01, -2.20388651e-01, -7.97302485e-01, ...,
           -5.44286060e+00, -5.84031153e+00, -5.41728687e+00],
          [ 1.38808489e-01, -2.11149678e-01, -3.94800723e-01, ...,
           -5.37926912e+00, -5.43685436e+00, -5.21165371e+00],
          [-7.18159378e-01, -1.80314258e-01, -1.25734091e-01, ...,
           -5.25551796e+00, -5.34799004e+00, -5.27835798e+00],
          ...,
          [ 4.05975133e-01, -2.33427271e-01, -3.45279127e-02, ...,
           -5.21699286e+00, -5.71105576e+00, -5.38287878e+00],
          [ 1.32289484e-01, -3.52178738e-02, -4.74997520e-01, ...,
           -5.21800804e+00, -5.32150841e+00, -5.12589645e+00],
          [-9.00482893e-01, -2.02055469e-01, -7.39011168e-01, ...,
           -5.39051723e+00, -5.66860580e+00, -5.02739811e+00]],

         ...,

         [[ 1.11851406e+00,  1.34091541e-01, -7.82647073e-01, ...,
           -5.50860977e+00, -6.16165876e+00, -5.63515663e+00],
          [-2.30284840e-01,  1.23895034e-01, -5.18168092e-01, ...,
           -5.27272749e+00, -6.11304426e+00, -5.64377594e+00],
          [-8.68184686e-01,  3.49291235e-01, -1.33422747e-01, ...,
           -5.44877815e+00, -6.34621906e+00, -5.87998104e+00],
          ...,
          [ 1.53782442e-01, -1.06271274e-01, -1.81500316e-02, ...,
           -5.30098152e+00, -5.95562840e+00, -5.47962713e+00],
          [ 1.58069551e-01,  9.66532901e-02, -4.89527106e-01, ...,
           -5.15649843e+00, -5.67873287e+00, -5.12191105e+00],
          [-8.76928508e-01,  2.25943759e-01, -8.43079448e-01, ...,
           -5.36193943e+00, -5.74983025e+00, -5.02586842e+00]],

         [[ 1.20568573e+00, -1.04791753e-01, -6.52817190e-01, ...,
           -5.46654987e+00, -6.04133177e+00, -5.39477062e+00],
          [ 9.95327458e-02, -2.53349990e-01, -3.85958076e-01, ...,
           -5.34690809e+00, -5.88741732e+00, -5.28640604e+00],
          [-4.82420892e-01,  2.63508856e-01, -7.83357471e-02, ...,
           -5.63745975e+00, -6.10578203e+00, -5.60612488e+00],
          ...,
          [ 1.38767928e-01, -9.58526134e-02, -7.68977404e-03, ...,
           -5.21965122e+00, -5.62968016e+00, -5.02352095e+00],
          [-5.36247939e-02, -1.52108157e-02, -4.16304529e-01, ...,
           -5.07022381e+00, -5.38883638e+00, -4.78220558e+00],
          [-8.41556668e-01, -3.26643735e-02, -8.20538044e-01, ...,
           -5.28739977e+00, -5.52591419e+00, -4.51837921e+00]],

         [[ 6.12165928e-01, -6.68815672e-01, -6.72047973e-01, ...,
           -5.61280060e+00, -6.00986814e+00, -5.44072056e+00],
          [ 1.97880611e-01, -1.26544392e+00, -3.85506332e-01, ...,
           -5.62314081e+00, -5.90508080e+00, -5.30341291e+00],
          [-3.28849971e-01, -8.77349675e-01, -4.33999002e-02, ...,
           -6.03714848e+00, -5.91169548e+00, -5.64734030e+00],
          ...,
          [ 2.90678442e-01, -1.19064116e+00, -2.12072730e-02, ...,
           -5.25232744e+00, -5.63611937e+00, -4.86030769e+00],
          [ 3.92162144e-01, -1.27825046e+00, -4.74981606e-01, ...,
           -5.09777975e+00, -5.42887831e+00, -4.51703644e+00],
          [-6.43159926e-01, -1.02756512e+00, -8.65118146e-01, ...,
           -5.45767164e+00, -5.54084158e+00, -4.53806067e+00]]],


        [[[-1.40306391e-02,  4.63948667e-01, -4.97904927e-01, ...,
           -5.17199659e+00, -5.30380821e+00, -4.67922306e+00],
          [ 1.63641766e-01,  1.73930597e+00, -5.30195236e-01, ...,
           -4.89507151e+00, -5.20778561e+00, -4.70921135e+00],
          [-2.48910278e-01,  1.74595046e+00, -5.90199113e-01, ...,
           -4.87020540e+00, -5.15122795e+00, -4.65494490e+00],
          ...,
          [ 1.03354409e-01,  1.61083627e+00, -5.26892304e-01, ...,
           -5.11858463e+00, -5.01647663e+00, -4.24402666e+00],
          [-2.24224180e-01,  1.54644287e+00, -6.71716809e-01, ...,
           -4.89773750e+00, -4.90058994e+00, -4.11995888e+00],
          [ 1.87354423e-02,  2.65842110e-01, -5.97165942e-01, ...,
           -4.89771557e+00, -5.12377405e+00, -4.25878143e+00]],

         [[ 8.69232476e-01,  6.23753726e-01, -6.68669283e-01, ...,
           -5.20534563e+00, -5.35349226e+00, -4.82974339e+00],
          [ 7.98686385e-01,  7.74117768e-01, -8.63505900e-01, ...,
           -4.80822229e+00, -5.22185612e+00, -4.71207762e+00],
          [-5.47325552e-01,  6.54311001e-01, -7.67403007e-01, ...,
           -4.59125328e+00, -5.27552319e+00, -4.83230639e+00],
          ...,
          [ 2.93599516e-01,  5.26799798e-01, -6.59623921e-01, ...,
           -4.87774277e+00, -5.25804186e+00, -4.76518011e+00],
          [-6.76950812e-01,  7.58918762e-01, -8.03217947e-01, ...,
           -4.57190037e+00, -4.99996948e+00, -4.43362188e+00],
          [-9.02082026e-01,  3.90693873e-01, -7.08806455e-01, ...,
           -4.75912189e+00, -5.17949867e+00, -4.40873289e+00]],

         [[ 9.04610038e-01, -3.83421294e-02, -7.72326529e-01, ...,
           -5.02961588e+00, -5.48226118e+00, -5.03989220e+00],
          [ 7.51640856e-01, -1.61999300e-01, -9.06606257e-01, ...,
           -5.06971836e+00, -5.43426323e+00, -5.02395535e+00],
          [-6.55927777e-01, -2.08627746e-01, -8.14871252e-01, ...,
           -4.98482132e+00, -5.48193359e+00, -5.13273525e+00],
          ...,
          [ 4.07857955e-01, -3.07808042e-01, -7.62642443e-01, ...,
           -4.93370056e+00, -5.68679619e+00, -5.21883345e+00],
          [-6.69862509e-01, -3.77479047e-02, -9.03460026e-01, ...,
           -4.94506216e+00, -5.36691666e+00, -4.95527697e+00],
          [-9.26899552e-01, -1.71688810e-01, -7.31634438e-01, ...,
           -4.97336197e+00, -5.38163900e+00, -4.73118591e+00]],

         ...,

         [[ 1.27541649e+00,  1.42385468e-01, -9.30784225e-01, ...,
           -5.13424492e+00, -5.72511101e+00, -5.25770092e+00],
          [ 7.74373710e-01,  1.87429354e-01, -9.23465431e-01, ...,
           -5.02927446e+00, -5.88193083e+00, -5.41232204e+00],
          [-8.00898373e-01,  2.73371518e-01, -8.19861710e-01, ...,
           -5.14672184e+00, -6.08063841e+00, -5.64399576e+00],
          ...,
          [ 2.53982633e-01, -1.03652969e-01, -7.68055439e-01, ...,
           -5.08776283e+00, -5.92881918e+00, -5.37374687e+00],
          [-8.17092359e-01,  1.18194208e-01, -9.24979687e-01, ...,
           -4.98560333e+00, -5.66494799e+00, -5.02895212e+00],
          [-1.19480848e+00,  1.57322705e-01, -8.76060963e-01, ...,
           -4.99654579e+00, -5.49651909e+00, -4.76026535e+00]],

         [[ 1.25082278e+00, -2.33649924e-01, -7.89908111e-01, ...,
           -5.12630177e+00, -5.62817240e+00, -5.07517242e+00],
          [ 8.22761953e-01, -7.49978006e-01, -8.56395721e-01, ...,
           -5.09824181e+00, -5.69581556e+00, -5.09481668e+00],
          [-5.92816591e-01, -5.20541310e-01, -7.64870524e-01, ...,
           -5.29259014e+00, -5.85100508e+00, -5.32544804e+00],
          ...,
          [ 1.15533367e-01, -5.94460130e-01, -7.24843562e-01, ...,
           -5.01855898e+00, -5.64935160e+00, -4.92993164e+00],
          [-8.24012816e-01, -4.76938814e-01, -8.77335072e-01, ...,
           -4.91434193e+00, -5.42822886e+00, -4.69843721e+00],
          [-9.75439787e-01, -2.78628141e-01, -8.00873518e-01, ...,
           -4.93028116e+00, -5.30495310e+00, -4.29904747e+00]],

         [[ 2.91152775e-01, -4.16881025e-01, -7.32878447e-01, ...,
           -5.18553686e+00, -5.54951906e+00, -5.02928066e+00],
          [ 5.22808015e-01, -1.75170684e+00, -7.83053398e-01, ...,
           -5.27742720e+00, -5.59383631e+00, -4.99233246e+00],
          [-1.94549486e-01, -1.64558470e+00, -6.80466950e-01, ...,
           -5.59759903e+00, -5.62683868e+00, -5.24174500e+00],
          ...,
          [ 1.29065588e-01, -2.05033112e+00, -7.14457870e-01, ...,
           -4.92152309e+00, -5.52737713e+00, -4.66256857e+00],
          [-3.58649731e-01, -1.95093369e+00, -8.23704958e-01, ...,
           -4.81920147e+00, -5.34615326e+00, -4.36220789e+00],
          [-4.02383879e-03, -5.71165323e-01, -7.30415046e-01, ...,
           -5.02056599e+00, -5.23202324e+00, -4.23245811e+00]]]]],
      dtype=float32), '950': array([[[[[-1.93465352e-01, -2.21953705e-01,  5.84107637e-03, ...,
           -5.62181187e+00, -5.49529171e+00, -5.37878799e+00],
          [-4.43443835e-01,  3.69891524e-01,  4.00879502e-01, ...,
           -5.50003004e+00, -5.74122667e+00, -5.09251022e+00],
          [ 2.00251073e-01,  7.06445575e-01,  1.52752745e+00, ...,
           -5.46649456e+00, -5.70688105e+00, -5.54405451e+00],
          ...,
          [ 9.48992074e-02,  1.43171763e+00,  1.55670774e+00, ...,
           -5.36140776e+00, -5.35724306e+00, -5.16760635e+00],
          [ 2.60346353e-01,  1.06808364e+00,  4.32577133e-01, ...,
           -5.08719492e+00, -5.47303724e+00, -4.76583910e+00],
          [-3.27699393e-01,  5.25045276e-01, -2.37915635e-01, ...,
           -5.18141699e+00, -5.93156242e+00, -5.24676132e+00]],

         [[ 2.85795271e-01,  9.75720584e-03, -9.66451764e-02, ...,
           -5.51572609e+00, -5.62627077e+00, -5.31100464e+00],
          [-4.21139032e-01,  6.78056628e-02,  4.53259647e-01, ...,
           -5.41883326e+00, -5.57777023e+00, -4.62851238e+00],
          [ 2.19334990e-01, -1.16325244e-01,  1.60120487e+00, ...,
           -4.99932194e+00, -5.44139528e+00, -5.16074514e+00],
          ...,
          [ 1.43538773e-01, -2.66781330e-01,  1.51833129e+00, ...,
           -5.04737616e+00, -4.99476910e+00, -4.71970749e+00],
          [ 2.63672233e-01, -5.42663038e-03,  4.82037246e-01, ...,
           -4.79474640e+00, -5.11443663e+00, -4.02937031e+00],
          [-6.85631037e-01,  9.61810052e-02, -7.96704888e-02, ...,
           -5.23228931e+00, -5.76113892e+00, -4.70573616e+00]],

         [[ 6.03383899e-01,  1.96720839e-01, -5.68034053e-02, ...,
           -5.72261286e+00, -5.81649256e+00, -5.56685734e+00],
          [-4.26719904e-01,  2.26040006e-01,  4.18491870e-01, ...,
           -5.35069275e+00, -5.76982069e+00, -5.05630922e+00],
          [-1.10801131e-01,  5.55815697e-02,  1.42665422e+00, ...,
           -5.07073975e+00, -5.64300776e+00, -5.37111044e+00],
          ...,
          [ 1.10877723e-01,  1.33267909e-01,  1.48646712e+00, ...,
           -5.09698391e+00, -4.97109795e+00, -4.73761654e+00],
          [ 2.84225702e-01, -1.78138405e-01,  4.09925163e-01, ...,
           -4.89960670e+00, -5.23788166e+00, -3.97943950e+00],
          [-6.67090297e-01, -2.29145646e-01, -1.14467025e-01, ...,
           -5.29821777e+00, -6.02617884e+00, -4.87800217e+00]],

         ...,

         [[ 5.97410083e-01,  5.81383258e-02,  1.24096870e-04, ...,
           -5.41529894e+00, -6.08564138e+00, -5.83440208e+00],
          [ 1.64211661e-01,  1.03525490e-01,  6.75848126e-01, ...,
           -5.53777933e+00, -5.87768936e+00, -5.56885433e+00],
          [-9.09172148e-02,  1.41609043e-01,  1.43151879e+00, ...,
           -5.59799814e+00, -5.87570858e+00, -5.73913670e+00],
          ...,
          [ 4.30153310e-01,  9.96060967e-02,  1.33947206e+00, ...,
           -5.09805393e+00, -5.36099291e+00, -5.32365561e+00],
          [-9.61564481e-02,  7.70402253e-02,  6.16361797e-01, ...,
           -5.07891941e+00, -5.16078281e+00, -4.49446106e+00],
          [-6.42711103e-01,  2.13392556e-01, -6.07350469e-02, ...,
           -5.39737272e+00, -5.54111671e+00, -4.58631802e+00]],

         [[ 4.86833304e-01,  2.17413306e-02, -9.75561142e-02, ...,
           -5.36647940e+00, -5.80889988e+00, -5.42411613e+00],
          [ 1.01447657e-01,  1.30080163e-01,  6.57034218e-01, ...,
           -5.31139421e+00, -5.61882687e+00, -5.19769001e+00],
          [-7.26552978e-02, -5.10334969e-04,  1.45690584e+00, ...,
           -5.21013641e+00, -5.68025875e+00, -5.59387398e+00],
          ...,
          [ 4.43495154e-01,  1.00250840e-02,  1.31490314e+00, ...,
           -5.05239439e+00, -5.18096972e+00, -4.78633738e+00],
          [ 1.86834931e-01, -1.32376105e-01,  4.98127043e-01, ...,
           -5.03353977e+00, -5.09255505e+00, -4.08209324e+00],
          [-5.44433653e-01,  2.70897150e-03, -1.57097459e-01, ...,
           -5.56357908e+00, -5.48637390e+00, -4.34542847e+00]],

         [[-2.80269265e-01, -4.60331500e-01,  1.45017833e-01, ...,
           -5.41827679e+00, -5.75269222e+00, -5.39216900e+00],
          [-9.97875333e-02, -1.00658631e+00,  6.77320719e-01, ...,
           -5.47934055e+00, -5.95097971e+00, -5.49101830e+00],
          [-7.77299479e-02, -9.42344248e-01,  1.57818770e+00, ...,
           -5.52024364e+00, -6.00518417e+00, -5.67284346e+00],
          ...,
          [ 1.41208678e-01, -6.28266692e-01,  1.46980476e+00, ...,
           -5.48208666e+00, -5.40077066e+00, -4.71672964e+00],
          [ 1.73838601e-01, -6.31869078e-01,  5.01515627e-01, ...,
           -5.42024946e+00, -5.35353279e+00, -4.20511723e+00],
          [-5.08968055e-01, -6.60838962e-01, -1.49529576e-01, ...,
           -5.76250553e+00, -5.65877962e+00, -4.80018806e+00]]],


        [[[-1.11420564e-02, -2.69534528e-01, -7.20554441e-02, ...,
           -5.57041883e+00, -5.55837345e+00, -5.46476507e+00],
          [-3.57667744e-01,  5.22704303e-01,  1.02497295e-01, ...,
           -5.45454931e+00, -5.83149290e+00, -5.14119387e+00],
          [ 3.09297115e-01,  6.98899746e-01,  3.23525429e-01, ...,
           -5.39230824e+00, -5.82597160e+00, -5.51163816e+00],
          ...,
          [ 1.16581038e-01,  1.42500865e+00,  3.32090706e-01, ...,
           -5.27473736e+00, -5.47863054e+00, -5.13572979e+00],
          [ 3.28862011e-01,  1.16704619e+00,  2.86541134e-03, ...,
           -4.97295380e+00, -5.54690552e+00, -4.81451797e+00],
          [-4.37456399e-01,  5.16354918e-01, -5.91672599e-01, ...,
           -5.08810949e+00, -5.98839331e+00, -5.33035183e+00]],

         [[ 4.80890900e-01, -4.46569137e-02, -2.36664847e-01, ...,
           -5.47437477e+00, -5.68439198e+00, -5.41438484e+00],
          [-2.25181147e-01,  1.58118308e-01, -7.91936070e-02, ...,
           -5.36739635e+00, -5.65562773e+00, -4.69787264e+00],
          [ 2.99016237e-01, -1.18488982e-01,  3.45746547e-01, ...,
           -4.95103455e+00, -5.55676746e+00, -5.14235592e+00],
          ...,
          [ 8.21684897e-02, -1.90129951e-01,  3.06809336e-01, ...,
           -4.95295525e+00, -5.11719942e+00, -4.71064949e+00],
          [ 2.62391955e-01,  4.84630764e-02, -2.33290732e-01, ...,
           -4.67420816e+00, -5.15659618e+00, -4.09233856e+00],
          [-8.01504672e-01,  5.01982868e-02, -5.68363369e-01, ...,
           -5.12194014e+00, -5.80647373e+00, -4.81293297e+00]],

         [[ 8.04740548e-01,  3.07142556e-01, -5.50350428e-01, ...,
           -5.66436100e+00, -5.86063910e+00, -5.63930750e+00],
          [-2.82880902e-01,  2.80491590e-01, -2.81440139e-01, ...,
           -5.31411552e+00, -5.82451296e+00, -5.10378170e+00],
          [-5.12364581e-02,  3.38265598e-02,  2.48227298e-01, ...,
           -5.05525780e+00, -5.71236849e+00, -5.36974907e+00],
          ...,
          [ 9.18334052e-02,  1.47741601e-01,  2.70992577e-01, ...,
           -5.03594589e+00, -5.06429195e+00, -4.77900124e+00],
          [ 2.77521729e-01, -1.35955244e-01, -3.15452904e-01, ...,
           -4.79282284e+00, -5.24766016e+00, -4.08215618e+00],
          [-8.07581365e-01, -1.02757663e-01, -6.53749287e-01, ...,
           -5.19831085e+00, -6.05044889e+00, -5.01614523e+00]],

         ...,

         [[ 9.20022368e-01,  1.60187989e-01, -3.96248400e-01, ...,
           -5.39652586e+00, -6.17254019e+00, -5.94753218e+00],
          [ 2.63043582e-01,  1.62000060e-01, -1.26195729e-01, ...,
           -5.53278017e+00, -6.01351357e+00, -5.64330912e+00],
          [-1.03454635e-01,  1.73174798e-01,  2.29251236e-01, ...,
           -5.58937550e+00, -6.02253008e+00, -5.79966354e+00],
          ...,
          [ 4.45328683e-01,  1.60993814e-01,  1.87986881e-01, ...,
           -5.05958843e+00, -5.51797104e+00, -5.39291954e+00],
          [-1.41765654e-01,  1.32216543e-01, -1.79280609e-01, ...,
           -5.00298786e+00, -5.26680994e+00, -4.58506393e+00],
          [-8.10207784e-01,  2.84275621e-01, -5.94925642e-01, ...,
           -5.31703281e+00, -5.61452866e+00, -4.69749975e+00]],

         [[ 8.25357676e-01,  2.80770361e-02, -3.81634384e-01, ...,
           -5.32370281e+00, -5.88664579e+00, -5.55031586e+00],
          [ 2.26982430e-01,  1.72496468e-01, -5.99979609e-02, ...,
           -5.28126907e+00, -5.74465942e+00, -5.25932693e+00],
          [-5.76529205e-02,  7.62119740e-02,  2.32364327e-01, ...,
           -5.19517994e+00, -5.83217001e+00, -5.62253094e+00],
          ...,
          [ 5.11049151e-01,  7.47781843e-02,  1.62167042e-01, ...,
           -5.00576973e+00, -5.32143879e+00, -4.83780098e+00],
          [ 8.74364674e-02, -7.94957653e-02, -2.10256338e-01, ...,
           -4.95201111e+00, -5.16336966e+00, -4.17863941e+00],
          [-8.31295788e-01,  9.69002098e-02, -6.00365579e-01, ...,
           -5.48051977e+00, -5.52058601e+00, -4.44985390e+00]],

         [[-3.07202861e-02, -3.12959254e-01, -1.41766876e-01, ...,
           -5.38406372e+00, -5.83270741e+00, -5.49626827e+00],
          [ 2.32865177e-02, -7.66126871e-01, -8.75834152e-02, ...,
           -5.47278690e+00, -6.05283356e+00, -5.50712395e+00],
          [ 1.40563883e-02, -7.47111142e-01,  2.83845305e-01, ...,
           -5.50423908e+00, -6.12978983e+00, -5.63299942e+00],
          ...,
          [ 1.19307347e-01, -5.09518623e-01,  2.54891872e-01, ...,
           -5.42500496e+00, -5.53218699e+00, -4.72461414e+00],
          [ 2.42900044e-01, -4.57763344e-01, -2.29120865e-01, ...,
           -5.38806629e+00, -5.41513777e+00, -4.24947071e+00],
          [-7.15375543e-01, -4.79000747e-01, -5.30226231e-01, ...,
           -5.72844028e+00, -5.71516323e+00, -4.89921284e+00]]],


        [[[-3.15766633e-02,  2.09784508e-01,  2.00502604e-01, ...,
           -5.36953163e+00, -5.60657978e+00, -5.33702660e+00],
          [-2.88385361e-01,  1.02823663e+00,  3.09952676e-01, ...,
           -5.23740482e+00, -5.79642868e+00, -5.00315380e+00],
          [ 2.63984948e-01,  1.41775346e+00,  3.82061005e-01, ...,
           -5.21754169e+00, -5.78662300e+00, -5.35760784e+00],
          ...,
          [ 6.48675784e-02,  1.99066639e+00,  3.69691074e-01, ...,
           -5.09198999e+00, -5.48603296e+00, -5.03796148e+00],
          [ 3.79304886e-01,  1.70746517e+00,  1.02169558e-01, ...,
           -4.74225521e+00, -5.55759430e+00, -4.69032669e+00],
          [-4.01753157e-01,  1.09989727e+00, -3.49105120e-01, ...,
           -4.84941483e+00, -6.01254177e+00, -5.20892048e+00]],

         [[ 6.02009356e-01,  1.08650774e-01, -5.27526662e-02, ...,
           -5.31180429e+00, -5.76126766e+00, -5.27538013e+00],
          [-9.22289491e-02,  3.56735528e-01,  3.41493264e-02, ...,
           -5.21479750e+00, -5.64217377e+00, -4.55857515e+00],
          [ 2.93886244e-01,  1.23080269e-01,  3.92799586e-01, ...,
           -4.80858898e+00, -5.57958412e+00, -5.03688049e+00],
          ...,
          [ 4.50273529e-02,  1.68437690e-01,  3.50134581e-01, ...,
           -4.78090858e+00, -5.14492941e+00, -4.66274357e+00],
          [ 2.22721279e-01,  1.89727291e-01, -2.04773486e-01, ...,
           -4.45053625e+00, -5.16060543e+00, -3.97294736e+00],
          [-9.36943889e-01,  2.20152736e-01, -5.29240727e-01, ...,
           -4.89368296e+00, -5.79831505e+00, -4.62899065e+00]],

         [[ 8.47276211e-01,  3.53950024e-01, -5.34205198e-01, ...,
           -5.56453466e+00, -5.92612743e+00, -5.40425253e+00],
          [-1.96687669e-01,  3.46433073e-01, -2.41004452e-01, ...,
           -5.23561096e+00, -5.87536621e+00, -4.92801428e+00],
          [-5.20737506e-02,  1.10384971e-01,  2.76112318e-01, ...,
           -4.98724508e+00, -5.80983067e+00, -5.27487326e+00],
          ...,
          [ 3.72782424e-02,  2.02322572e-01,  3.10145468e-01, ...,
           -4.91511679e+00, -5.17028332e+00, -4.72862673e+00],
          [ 2.53561288e-01, -1.14878260e-01, -2.84251809e-01, ...,
           -4.62202168e+00, -5.29942989e+00, -3.95647550e+00],
          [-8.34612250e-01, -8.02832767e-02, -6.25530362e-01, ...,
           -4.99181175e+00, -6.05996704e+00, -4.81776953e+00]],

         ...,

         [[ 1.02227390e+00,  1.11384615e-01, -3.73507142e-01, ...,
           -5.29357624e+00, -6.21883583e+00, -5.76373816e+00],
          [ 2.92776287e-01,  9.68637317e-02, -8.29644203e-02, ...,
           -5.42930174e+00, -6.07650423e+00, -5.53760433e+00],
          [-8.10935572e-02,  1.21018529e-01,  2.75461555e-01, ...,
           -5.50554180e+00, -6.10432529e+00, -5.74203300e+00],
          ...,
          [ 4.10006583e-01,  1.03878856e-01,  2.31676355e-01, ...,
           -4.96559763e+00, -5.67210770e+00, -5.32181978e+00],
          [-1.75826892e-01,  4.86417413e-02, -1.41519934e-01, ...,
           -4.89056683e+00, -5.40405989e+00, -4.49140167e+00],
          [-8.14634144e-01,  2.13825494e-01, -5.71962774e-01, ...,
           -5.15844917e+00, -5.70908451e+00, -4.55906868e+00]],

         [[ 9.69749451e-01, -1.26692891e-01, -3.12882185e-01, ...,
           -5.20141268e+00, -5.92081928e+00, -5.40241957e+00],
          [ 2.79884934e-01,  4.22570109e-03, -1.98169351e-02, ...,
           -5.14808798e+00, -5.77640152e+00, -5.15084505e+00],
          [-6.13857843e-02, -5.65104336e-02,  2.79866815e-01, ...,
           -5.08764172e+00, -5.90582561e+00, -5.56695938e+00],
          ...,
          [ 4.52727288e-01, -2.03113362e-01,  2.09662527e-01, ...,
           -4.87767887e+00, -5.46468115e+00, -4.79806232e+00],
          [-1.29365036e-02, -2.37815693e-01, -1.53324753e-01, ...,
           -4.82727766e+00, -5.26305294e+00, -4.08378267e+00],
          [-8.80445063e-01, -8.34185481e-02, -5.62957406e-01, ...,
           -5.30204439e+00, -5.55474854e+00, -4.29933119e+00]],

         [[-3.76985036e-02, -7.52711177e-01,  3.65228020e-02, ...,
           -5.22269201e+00, -5.89974451e+00, -5.40562725e+00],
          [ 2.66162958e-02, -1.42224503e+00, -2.38436684e-02, ...,
           -5.32349586e+00, -6.04289913e+00, -5.40607548e+00],
          [ 3.55589278e-02, -1.45156491e+00,  3.39306086e-01, ...,
           -5.34667587e+00, -6.10001040e+00, -5.57207823e+00],
          ...,
          [ 8.53466149e-03, -1.43204999e+00,  3.62344682e-01, ...,
           -5.25129700e+00, -5.59232903e+00, -4.71450472e+00],
          [ 7.98127949e-02, -1.48964679e+00, -9.95873362e-02, ...,
           -5.21076107e+00, -5.45152378e+00, -4.19667816e+00],
          [-6.61481321e-01, -1.07947767e+00, -3.20786327e-01, ...,
           -5.54840422e+00, -5.74757481e+00, -4.80275059e+00]]]]],
      dtype=float32), '937': array([[[[[ 2.10083008e-01,  8.24584961e-02,  1.16699219e-01, ...,
           -5.69921875e+00, -6.55078125e+00, -4.96093750e+00],
          [ 1.08642578e-01,  3.82080078e-01,  1.37304688e+00, ...,
           -5.70312500e+00, -6.34765625e+00, -5.16015625e+00],
          [-6.04003906e-01,  2.23022461e-01,  2.38476562e+00, ...,
           -5.87109375e+00, -6.24609375e+00, -4.68750000e+00],
          ...,
          [ 6.71386719e-01,  8.51074219e-01,  2.47656250e+00, ...,
           -5.73046875e+00, -6.14843750e+00, -4.46484375e+00],
          [ 3.69873047e-02,  8.54492188e-01,  1.38378906e+00, ...,
           -5.73828125e+00, -6.44921875e+00, -5.37500000e+00],
          [ 1.61743164e-01,  2.60009766e-01, -1.15356445e-01, ...,
           -5.82812500e+00, -6.71093750e+00, -5.33203125e+00]],

         [[ 3.21533203e-01, -1.89819336e-01,  4.41284180e-02, ...,
           -5.73828125e+00, -6.40625000e+00, -5.03125000e+00],
          [ 2.45239258e-01,  1.38092041e-02,  1.53027344e+00, ...,
           -5.45312500e+00, -6.01562500e+00, -5.08203125e+00],
          [-7.30957031e-01,  5.30700684e-02,  2.42578125e+00, ...,
           -5.41796875e+00, -5.78906250e+00, -4.60156250e+00],
          ...,
          [ 9.09667969e-01,  2.65869141e-01,  2.51171875e+00, ...,
           -5.36328125e+00, -5.81640625e+00, -4.35937500e+00],
          [ 8.72802734e-02,  4.56542969e-01,  1.29101562e+00, ...,
           -5.39453125e+00, -6.02343750e+00, -5.12109375e+00],
          [-2.74169922e-01,  7.25097656e-02,  9.14306641e-02, ...,
           -5.67187500e+00, -6.48437500e+00, -5.23046875e+00]],

         [[ 2.44873047e-01, -2.60009766e-01, -2.84423828e-02, ...,
           -5.90234375e+00, -6.46093750e+00, -5.20703125e+00],
          [ 2.90527344e-01,  5.24902344e-02,  1.52734375e+00, ...,
           -5.57421875e+00, -6.03906250e+00, -5.32421875e+00],
          [-6.46484375e-01,  1.33178711e-01,  2.23046875e+00, ...,
           -5.50390625e+00, -5.87890625e+00, -4.86328125e+00],
          ...,
          [ 5.56152344e-01, -1.36962891e-01,  2.58984375e+00, ...,
           -5.32421875e+00, -5.78515625e+00, -4.20312500e+00],
          [ 6.39648438e-02, -1.42089844e-01,  1.25000000e+00, ...,
           -5.44140625e+00, -6.12890625e+00, -4.89843750e+00],
          [-5.91308594e-01, -1.91284180e-01,  2.25952148e-01, ...,
           -5.60937500e+00, -6.57421875e+00, -4.99218750e+00]],

         ...,

         [[ 4.14306641e-01,  1.72729492e-01,  6.54907227e-02, ...,
           -5.47265625e+00, -6.24218750e+00, -5.13671875e+00],
          [ 3.66943359e-01,  4.13818359e-02,  1.57519531e+00, ...,
           -5.32031250e+00, -5.85937500e+00, -5.30859375e+00],
          [-5.56640625e-01, -7.91015625e-02,  2.54101562e+00, ...,
           -5.16015625e+00, -5.56250000e+00, -4.95703125e+00],
          ...,
          [ 6.00585938e-01,  2.02636719e-01,  2.46679688e+00, ...,
           -5.33593750e+00, -5.51953125e+00, -4.55468750e+00],
          [ 2.28393555e-01,  2.31323242e-01,  1.13671875e+00, ...,
           -5.38671875e+00, -5.78125000e+00, -4.88671875e+00],
          [-3.22532654e-03,  1.58569336e-01, -1.67236328e-01, ...,
           -5.58203125e+00, -6.14453125e+00, -4.70312500e+00]],

         [[ 4.71923828e-01, -7.00683594e-02,  1.58081055e-01, ...,
           -5.70312500e+00, -6.19140625e+00, -5.35546875e+00],
          [ 3.76892090e-02,  1.64489746e-02,  1.30468750e+00, ...,
           -5.55468750e+00, -5.91015625e+00, -5.44531250e+00],
          [-6.20117188e-01, -1.22070312e-02,  2.49804688e+00, ...,
           -5.48828125e+00, -5.67968750e+00, -5.05859375e+00],
          ...,
          [ 5.09277344e-01,  1.91772461e-01,  2.50781250e+00, ...,
           -5.47656250e+00, -5.49609375e+00, -4.40625000e+00],
          [ 1.67358398e-01,  1.30493164e-01,  1.15234375e+00, ...,
           -5.55078125e+00, -5.77734375e+00, -5.09375000e+00],
          [-2.36694336e-01,  8.77075195e-02,  3.36608887e-02, ...,
           -5.76171875e+00, -6.04296875e+00, -4.83593750e+00]],

         [[ 7.00195312e-01, -1.67358398e-01,  2.09594727e-01, ...,
           -5.81640625e+00, -6.50781250e+00, -5.26562500e+00],
          [-1.60522461e-01, -1.08642578e-01,  1.06738281e+00, ...,
           -5.84375000e+00, -6.32031250e+00, -5.30859375e+00],
          [-3.83789062e-01, -1.67114258e-01,  2.09960938e+00, ...,
           -5.82031250e+00, -6.14062500e+00, -4.95312500e+00],
          ...,
          [ 5.28320312e-01, -1.49414062e-01,  2.10546875e+00, ...,
           -5.85546875e+00, -5.93750000e+00, -4.29296875e+00],
          [ 1.54541016e-01, -4.74243164e-02,  1.08007812e+00, ...,
           -5.92968750e+00, -6.27734375e+00, -5.12500000e+00],
          [-3.05908203e-01, -2.23388672e-02,  2.54150391e-01, ...,
           -5.97265625e+00, -6.48828125e+00, -5.05468750e+00]]],


        [[[ 2.23144531e-01,  1.97509766e-01, -9.16748047e-02, ...,
           -5.36328125e+00, -6.26171875e+00, -4.77343750e+00],
          [ 2.44750977e-01,  5.42480469e-01,  6.31347656e-01, ...,
           -5.37890625e+00, -6.05468750e+00, -4.98437500e+00],
          [-2.68310547e-01,  4.58251953e-01,  1.40136719e+00, ...,
           -5.52343750e+00, -5.96875000e+00, -4.67187500e+00],
          ...,
          [ 2.06787109e-01,  1.04003906e+00,  1.44042969e+00, ...,
           -5.49218750e+00, -5.94140625e+00, -4.53906250e+00],
          [-2.43774414e-01,  9.55566406e-01,  6.22558594e-01, ...,
           -5.57421875e+00, -6.27343750e+00, -5.28515625e+00],
          [ 6.83212280e-03,  4.51660156e-01, -2.23999023e-01, ...,
           -5.56250000e+00, -6.53125000e+00, -5.21093750e+00]],

         [[ 4.33593750e-01, -5.44433594e-02, -3.03955078e-01, ...,
           -5.46093750e+00, -6.16015625e+00, -4.81250000e+00],
          [ 3.70117188e-01,  1.14822388e-02,  6.60156250e-01, ...,
           -5.26953125e+00, -5.81640625e+00, -4.89453125e+00],
          [-6.90307617e-02,  2.75573730e-02,  1.43652344e+00, ...,
           -5.25000000e+00, -5.61718750e+00, -4.62890625e+00],
          ...,
          [ 1.51977539e-01,  2.84423828e-01,  1.42871094e+00, ...,
           -5.27734375e+00, -5.67578125e+00, -4.42187500e+00],
          [-2.47192383e-01,  4.74609375e-01,  5.91308594e-01, ...,
           -5.34375000e+00, -5.88671875e+00, -5.01953125e+00],
          [-5.43457031e-01,  1.06506348e-01, -2.60009766e-01, ...,
           -5.51562500e+00, -6.34765625e+00, -5.08203125e+00]],

         [[ 3.24462891e-01,  6.80541992e-02, -3.61083984e-01, ...,
           -5.62890625e+00, -6.26171875e+00, -5.04687500e+00],
          [ 3.59619141e-01,  2.93212891e-01,  6.54785156e-01, ...,
           -5.39453125e+00, -5.89062500e+00, -5.20312500e+00],
          [-2.59033203e-01,  3.86718750e-01,  1.25683594e+00, ...,
           -5.33593750e+00, -5.71875000e+00, -4.91796875e+00],
          ...,
          [-1.99462891e-01,  4.80651855e-02,  1.42187500e+00, ...,
           -5.16406250e+00, -5.58593750e+00, -4.30078125e+00],
          [-1.51855469e-01, -3.01971436e-02,  5.29785156e-01, ...,
           -5.33984375e+00, -5.98828125e+00, -4.80859375e+00],
          [-8.03710938e-01, -1.69086456e-03, -1.79321289e-01, ...,
           -5.39062500e+00, -6.45312500e+00, -4.85156250e+00]],

         ...,

         [[ 5.36621094e-01, -6.71386719e-02, -2.90527344e-01, ...,
           -5.17578125e+00, -6.08203125e+00, -5.06250000e+00],
          [ 4.61181641e-01, -2.76123047e-01,  6.84082031e-01, ...,
           -5.14843750e+00, -5.75000000e+00, -5.28125000e+00],
          [-5.63354492e-02, -3.58642578e-01,  1.34863281e+00, ...,
           -5.02734375e+00, -5.46484375e+00, -5.14843750e+00],
          ...,
          [ 8.38623047e-02, -2.12280273e-01,  1.31054688e+00, ...,
           -5.20312500e+00, -5.41015625e+00, -4.69140625e+00],
          [ 6.56738281e-02, -1.32934570e-01,  4.50195312e-01, ...,
           -5.24218750e+00, -5.62890625e+00, -4.81640625e+00],
          [-1.92382812e-01, -2.29492188e-02, -4.44580078e-01, ...,
           -5.31640625e+00, -5.99609375e+00, -4.59765625e+00]],

         [[ 5.25878906e-01, -4.00390625e-01, -2.44873047e-01, ...,
           -5.46484375e+00, -5.98437500e+00, -5.20312500e+00],
          [ 9.66186523e-02, -2.91503906e-01,  5.04882812e-01, ...,
           -5.36718750e+00, -5.73828125e+00, -5.34375000e+00],
          [-1.91772461e-01, -1.48925781e-01,  1.31445312e+00, ...,
           -5.31640625e+00, -5.51562500e+00, -5.14062500e+00],
          ...,
          [ 1.18530273e-01,  1.05285645e-01,  1.32910156e+00, ...,
           -5.31250000e+00, -5.30468750e+00, -4.43750000e+00],
          [-1.70745850e-02, -4.76074219e-03,  4.61914062e-01, ...,
           -5.43750000e+00, -5.57812500e+00, -4.94140625e+00],
          [-4.97070312e-01, -8.63037109e-02, -2.26562500e-01, ...,
           -5.52343750e+00, -5.84375000e+00, -4.68359375e+00]],

         [[ 6.45507812e-01, -3.24951172e-01, -1.20605469e-01, ...,
           -5.51953125e+00, -6.29296875e+00, -5.18359375e+00],
          [ 5.12084961e-02, -4.41650391e-01,  4.76318359e-01, ...,
           -5.60937500e+00, -6.14062500e+00, -5.25000000e+00],
          [-2.35473633e-01, -4.59228516e-01,  1.43261719e+00, ...,
           -5.55859375e+00, -5.94140625e+00, -5.03515625e+00],
          ...,
          [ 2.70996094e-01, -4.09912109e-01,  1.42480469e+00, ...,
           -5.59765625e+00, -5.69921875e+00, -4.32031250e+00],
          [-6.82373047e-02, -3.44726562e-01,  4.59472656e-01, ...,
           -5.73437500e+00, -6.05859375e+00, -5.02734375e+00],
          [-4.61425781e-01, -2.34527588e-02,  9.21630859e-02, ...,
           -5.68359375e+00, -6.28125000e+00, -4.96875000e+00]]],


        [[[ 3.67187500e-01,  1.64306641e-01, -4.44580078e-01, ...,
           -5.19140625e+00, -6.19531250e+00, -4.89843750e+00],
          [ 3.09326172e-01,  4.87792969e-01, -1.92504883e-01, ...,
           -5.20703125e+00, -6.03515625e+00, -4.98437500e+00],
          [-1.46240234e-01,  4.16992188e-01,  2.56591797e-01, ...,
           -5.31640625e+00, -5.96093750e+00, -4.66015625e+00],
          ...,
          [ 2.98095703e-01,  9.89257812e-01,  2.97363281e-01, ...,
           -5.28515625e+00, -5.94531250e+00, -4.55859375e+00],
          [-1.82983398e-01,  8.99414062e-01, -1.76269531e-01, ...,
           -5.43359375e+00, -6.28515625e+00, -5.28906250e+00],
          [-1.97021484e-01,  4.14550781e-01, -5.83007812e-01, ...,
           -5.40234375e+00, -6.46875000e+00, -5.32812500e+00]],

         [[ 7.17285156e-01, -3.91235352e-02, -7.18750000e-01, ...,
           -5.30859375e+00, -6.09375000e+00, -4.96484375e+00],
          [ 3.47167969e-01,  4.57458496e-02, -1.81030273e-01, ...,
           -5.12500000e+00, -5.80859375e+00, -4.91015625e+00],
          [ 7.17163086e-02,  6.97631836e-02,  3.27392578e-01, ...,
           -5.06640625e+00, -5.62500000e+00, -4.64843750e+00],
          ...,
          [ 1.87133789e-01,  3.23242188e-01,  3.21289062e-01, ...,
           -5.10156250e+00, -5.68750000e+00, -4.46484375e+00],
          [-2.50976562e-01,  4.97070312e-01, -1.82128906e-01, ...,
           -5.21484375e+00, -5.90234375e+00, -5.06640625e+00],
          [-8.45703125e-01,  1.05163574e-01, -6.49902344e-01, ...,
           -5.37500000e+00, -6.30468750e+00, -5.20312500e+00]],

         [[ 6.49902344e-01,  3.51257324e-02, -7.65625000e-01, ...,
           -5.50781250e+00, -6.17187500e+00, -5.19140625e+00],
          [ 3.32275391e-01,  2.75878906e-01, -1.96044922e-01, ...,
           -5.29687500e+00, -5.87109375e+00, -5.21484375e+00],
          [-1.45751953e-01,  3.76953125e-01,  2.27783203e-01, ...,
           -5.20703125e+00, -5.68750000e+00, -4.93359375e+00],
          ...,
          [-1.67480469e-01,  5.08422852e-02,  2.82226562e-01, ...,
           -5.00781250e+00, -5.51171875e+00, -4.30859375e+00],
          [-1.39160156e-01, -6.20727539e-02, -2.38647461e-01, ...,
           -5.22265625e+00, -5.92968750e+00, -4.82421875e+00],
          [-1.07324219e+00, -3.95812988e-02, -6.61132812e-01, ...,
           -5.25390625e+00, -6.34375000e+00, -4.93359375e+00]],

         ...,

         [[ 8.77441406e-01, -1.44531250e-01, -7.11914062e-01, ...,
           -5.06250000e+00, -6.01562500e+00, -5.26171875e+00],
          [ 4.23339844e-01, -3.40820312e-01, -1.61376953e-01, ...,
           -5.08203125e+00, -5.74609375e+00, -5.35156250e+00],
          [ 1.30920410e-02, -4.09423828e-01,  2.64404297e-01, ...,
           -4.93359375e+00, -5.47656250e+00, -5.22265625e+00],
          ...,
          [ 8.58154297e-02, -2.70751953e-01,  2.07153320e-01, ...,
           -5.08593750e+00, -5.38671875e+00, -4.77734375e+00],
          [ 9.08813477e-02, -2.00805664e-01, -3.19580078e-01, ...,
           -5.14062500e+00, -5.58203125e+00, -4.90234375e+00],
          [-4.80712891e-01, -8.83178711e-02, -8.37402344e-01, ...,
           -5.17578125e+00, -5.89062500e+00, -4.75781250e+00]],

         [[ 8.30566406e-01, -4.82666016e-01, -6.94335938e-01, ...,
           -5.32421875e+00, -5.92968750e+00, -5.36718750e+00],
          [ 9.72900391e-02, -3.76220703e-01, -2.79541016e-01, ...,
           -5.27343750e+00, -5.73046875e+00, -5.41796875e+00],
          [-1.05957031e-01, -2.26806641e-01,  2.45849609e-01, ...,
           -5.18359375e+00, -5.52734375e+00, -5.19531250e+00],
          ...,
          [ 1.43188477e-01,  5.05981445e-02,  2.20825195e-01, ...,
           -5.16015625e+00, -5.28906250e+00, -4.50390625e+00],
          [ 3.93371582e-02, -7.44018555e-02, -2.94921875e-01, ...,
           -5.32812500e+00, -5.55468750e+00, -5.01562500e+00],
          [-7.36816406e-01, -1.55517578e-01, -6.75292969e-01, ...,
           -5.36328125e+00, -5.76562500e+00, -4.85156250e+00]],

         [[ 7.72460938e-01, -3.65966797e-01, -5.47851562e-01, ...,
           -5.40234375e+00, -6.23828125e+00, -5.32812500e+00],
          [ 1.62353516e-01, -4.95117188e-01, -2.76367188e-01, ...,
           -5.52734375e+00, -6.13671875e+00, -5.30859375e+00],
          [-1.28906250e-01, -4.94628906e-01,  3.01269531e-01, ...,
           -5.41015625e+00, -5.94531250e+00, -5.03906250e+00],
          ...,
          [ 3.26416016e-01, -4.29443359e-01,  3.15185547e-01, ...,
           -5.44531250e+00, -5.67968750e+00, -4.33593750e+00],
          [-4.35180664e-02, -3.84277344e-01, -2.65869141e-01, ...,
           -5.63281250e+00, -6.03125000e+00, -5.06640625e+00],
          [-5.20996094e-01, -6.34155273e-02, -4.65576172e-01, ...,
           -5.55078125e+00, -6.21484375e+00, -5.08593750e+00]]]]],
      dtype=float32)}

The whole script is in attachment:
coreml-mlmodel.ipynb.zip

[github-actions]

👋 Hello @tommypham, thank you for your interest in YOLOv5 🚀! Please visit our ⭐️ Tutorials to get started, where you can find quickstart guides for simple tasks like Custom Data Training all the way to advanced concepts like Hyperparameter Evolution.

If this is a 🐛 Bug Report, please provide screenshots and minimum viable code to reproduce your issue, otherwise we can not help you.

If this is a custom training ❓ Question, please provide as much information as possible, including dataset images, training logs, screenshots, and a public link to online W&B logging if available.

For business inquiries or professional support requests please visit https://ultralytics.com or email Glenn Jocher at [email protected].

Requirements

Python>=3.6.0 with all requirements.txt installed including PyTorch>=1.7. To get started:

$ git clone https://github.com/ultralytics/yolov5
$ cd yolov5
$ pip install -r requirements.txt

Environments

YOLOv5 may be run in any of the following up-to-date verified environments (with all dependencies including CUDA/CUDNN, Python and PyTorch preinstalled):

• Google Colab and Kaggle notebooks with free GPU:
• Google Cloud Deep Learning VM. See GCP Quickstart Guide
• Amazon Deep Learning AMI. See AWS Quickstart Guide
• Docker Image. See Docker Quickstart Guide

Status

If this badge is green, all YOLOv5 GitHub Actions Continuous Integration (CI) tests are currently passing. CI tests verify correct operation of YOLOv5 training (train.py), validation (val.py), inference (detect.py) and export (export.py) on MacOS, Windows, and Ubuntu every 24 hours and on every commit.

[pytholic]

@tommypham did you figure it out mate?

[tommypham]

@rajahaseeb147 : I think we should apply sigmoid function to interpret it to real X, Y , width , height. Reference at : #343 (comment)

[github-actions]

👋 Hello, this issue has been automatically marked as stale because it has not had recent activity. Please note it will be closed if no further activity occurs.

Access additional YOLOv5 🚀 resources:

• Wiki – https://github.com/ultralytics/yolov5/wiki
• Tutorials – https://docs.ultralytics.com/yolov5
• Docs – https://docs.ultralytics.com

Access additional Ultralytics ⚡ resources:

• Ultralytics HUB – https://ultralytics.com/hub
• Vision API – https://ultralytics.com/yolov5
• About Us – https://ultralytics.com/about
• Join Our Team – https://ultralytics.com/work
• Contact Us – https://ultralytics.com/contact

Feel free to inform us of any other issues you discover or feature requests that come to mind in the future. Pull Requests (PRs) are also always welcomed!

Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐!

[glenn-jocher]

@tommypham @pytholic good news 😃! Your original issue may now be fixed ✅ in PR #6195. This PR adds support for YOLOv5 CoreML inference.

!python export.py --weights yolov5s.pt --include coreml  # CoreML export
!python detect.py --weights yolov5s.mlmodel  # CoreML inference (MacOS-only)
!python val.py --weights yolov5s.mlmodel  # CoreML validation (MacOS-only)

model = torch.hub.load('ultralytics/yolov5', 'custom', 'yolov5s.mlmodel')  # CoreML PyTorch Hub model

To receive this update:

• Git – git pull from within your yolov5/ directory or git clone https://github.com/ultralytics/yolov5 again
• PyTorch Hub – Force-reload model = torch.hub.load('ultralytics/yolov5', 'yolov5s', force_reload=True)
• Notebooks – View updated notebooks
• Docker – sudo docker pull ultralytics/yolov5:latest to update your image

Thank you for spotting this issue and informing us of the problem. Please let us know if this update resolves the issue for you, and feel free to inform us of any other issues you discover or feature requests that come to mind. Happy trainings with YOLOv5 🚀!