ROMP icon indicating copy to clipboard operation
ROMP copied to clipboard

[simple-romp] argument '--show_largest' not working properly?

Open vivi90 opened this issue 2 years ago • 1 comments

Tested with the following command:

romp -t -sc=3 --onnx --show_largest --mode=video -i "E:\Projekte\romp-test\test.mp4" --calc_smpl --render_mesh --save_video -o "E:\Projekte\romp-test\test_result.mp4"

And get the following result at frame 215 of this video: Frame 215

Why does it evaluate not only the largest person (--show_largest), according to the README?: [..] to show the largest person only (remove the small subjects in background) [..]

The data:

{
  'cam': array([
    [0.6815796, -0.7091072, 0.5703686],
    [1.1357857, -0.18335153, 0.77637017]
  ], dtype = float32),
  'global_orient': array([
    [2.991697, -0.11487218, 0.02521912],
    [3.1261547, 0.06689704, -0.28957558]
  ], dtype = float32),
  'body_pose': array([
    [-5.85280538e-01, -2.17623878e-02, 1.04836568e-01,
      -5.94840944e-01, 2.05031112e-02, -4.04515043e-02,
      4.90758419e-01, -1.56615092e-03, -1.22337053e-02,
      1.18982077e+00, 6.25830796e-03, -1.33860081e-01,
      1.03421855e+00, -4.33490239e-02, 9.75086764e-02,
      8.08686297e-03, -1.64354779e-02, -1.56924091e-02,
      -5.29559441e-02, 1.50208578e-01, -8.31045508e-02,
      -6.23615980e-02, -1.55124173e-01, 8.85685012e-02,
      3.80534641e-02, -1.49868811e-02, -1.34220216e-02,
      -2.74092108e-01, 9.23865065e-02, 1.74044952e-01,
      -2.23217666e-01, -2.57145371e-02, -1.95539817e-01,
      -1.50648043e-01, 2.01505367e-02, -2.51643322e-02,
      3.70577164e-02, -2.34452099e-01, -3.42204958e-01,
      2.55778432e-02, 2.65333027e-01, 2.82382011e-01,
      8.90259165e-03, 5.71487285e-03, 1.26959458e-02,
      9.14356261e-02, -3.04939717e-01, -9.25578892e-01,
      1.75735921e-01, 3.51879328e-01, 9.20472264e-01,
      6.75733909e-02, -5.05667686e-01, 6.31040707e-02,
      2.48549446e-01, 8.01332653e-01, -9.67753083e-02,
      7.34215826e-02, -6.02314919e-02, 1.28485382e-01,
      5.53916991e-02, 8.01604688e-02, -1.09875299e-01,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00
    ],
    [-4.36773539e-01, -4.78146598e-02, 1.86988581e-02,
      -4.45884168e-01, 6.41934425e-02, -3.56641151e-02,
      3.63900691e-01, -1.11775724e-02, 1.87774356e-02,
      9.40925658e-01, 1.05378553e-02, -1.03348374e-01,
      8.98670733e-01, 8.93227570e-03, 3.17471549e-02,
      -1.37289772e-02, -2.22832896e-02, 1.29298996e-02,
      -5.16718961e-02, 1.28186598e-01, -6.89635724e-02,
      -5.80723621e-02, -1.42094582e-01, 1.20365016e-01,
      2.41966341e-02, -1.15594221e-02, 1.08879823e-02,
      -2.70833313e-01, 1.29253507e-01, 1.61255777e-01,
      -2.09726378e-01, -1.71624217e-02, -2.16740996e-01,
      -1.97537243e-01, -1.50388693e-02, 1.52160926e-02,
      -9.75993183e-03, -2.94516653e-01, -2.75821269e-01,
      -4.38967049e-02, 3.35539162e-01, 1.97340801e-01,
      2.59927511e-02, 4.17219533e-04, 1.58897098e-02,
      8.48777741e-02, -4.73912209e-01, -9.98008311e-01,
      9.62874740e-02, 5.07905126e-01, 9.15271461e-01,
      1.01814926e-01, -7.08249331e-01, 1.66313887e-01,
      9.26086754e-02, 9.67943072e-01, -2.75810599e-01,
      -6.72952756e-02, -8.33046138e-02, 1.30613551e-01,
      -2.68768333e-02, 9.00805369e-02, -1.20457843e-01,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00
    ]
  ], dtype = float32),
  'smpl_betas': array([
    [-5.5625755e-03, -6.6001648e-03, 2.7066749e-04, 7.3520839e-04,
      1.0587471e-03, -1.6262289e-05, -1.6828703e-03, -1.9729475e-04,
      1.1870718e-03, -1.1305653e-03
    ],
    [9.1921221e-03, -2.7188463e-03, 1.5820982e-04, -3.2527829e-03,
      1.9372397e-05, -9.4282208e-05, -1.2378157e-03, 3.5660830e-04,
      7.2848942e-04, -5.3632428e-04
    ]
  ], dtype = float32),
  'smpl_thetas': array([
    [2.99169707e+00, -1.14872180e-01, 2.52191164e-02,
      -5.85280538e-01, -2.17623878e-02, 1.04836568e-01,
      -5.94840944e-01, 2.05031112e-02, -4.04515043e-02,
      4.90758419e-01, -1.56615092e-03, -1.22337053e-02,
      1.18982077e+00, 6.25830796e-03, -1.33860081e-01,
      1.03421855e+00, -4.33490239e-02, 9.75086764e-02,
      8.08686297e-03, -1.64354779e-02, -1.56924091e-02,
      -5.29559441e-02, 1.50208578e-01, -8.31045508e-02,
      -6.23615980e-02, -1.55124173e-01, 8.85685012e-02,
      3.80534641e-02, -1.49868811e-02, -1.34220216e-02,
      -2.74092108e-01, 9.23865065e-02, 1.74044952e-01,
      -2.23217666e-01, -2.57145371e-02, -1.95539817e-01,
      -1.50648043e-01, 2.01505367e-02, -2.51643322e-02,
      3.70577164e-02, -2.34452099e-01, -3.42204958e-01,
      2.55778432e-02, 2.65333027e-01, 2.82382011e-01,
      8.90259165e-03, 5.71487285e-03, 1.26959458e-02,
      9.14356261e-02, -3.04939717e-01, -9.25578892e-01,
      1.75735921e-01, 3.51879328e-01, 9.20472264e-01,
      6.75733909e-02, -5.05667686e-01, 6.31040707e-02,
      2.48549446e-01, 8.01332653e-01, -9.67753083e-02,
      7.34215826e-02, -6.02314919e-02, 1.28485382e-01,
      5.53916991e-02, 8.01604688e-02, -1.09875299e-01,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00
    ],
    [3.12615466e+00, 6.68970421e-02, -2.89575577e-01,
      -4.36773539e-01, -4.78146598e-02, 1.86988581e-02,
      -4.45884168e-01, 6.41934425e-02, -3.56641151e-02,
      3.63900691e-01, -1.11775724e-02, 1.87774356e-02,
      9.40925658e-01, 1.05378553e-02, -1.03348374e-01,
      8.98670733e-01, 8.93227570e-03, 3.17471549e-02,
      -1.37289772e-02, -2.22832896e-02, 1.29298996e-02,
      -5.16718961e-02, 1.28186598e-01, -6.89635724e-02,
      -5.80723621e-02, -1.42094582e-01, 1.20365016e-01,
      2.41966341e-02, -1.15594221e-02, 1.08879823e-02,
      -2.70833313e-01, 1.29253507e-01, 1.61255777e-01,
      -2.09726378e-01, -1.71624217e-02, -2.16740996e-01,
      -1.97537243e-01, -1.50388693e-02, 1.52160926e-02,
      -9.75993183e-03, -2.94516653e-01, -2.75821269e-01,
      -4.38967049e-02, 3.35539162e-01, 1.97340801e-01,
      2.59927511e-02, 4.17219533e-04, 1.58897098e-02,
      8.48777741e-02, -4.73912209e-01, -9.98008311e-01,
      9.62874740e-02, 5.07905126e-01, 9.15271461e-01,
      1.01814926e-01, -7.08249331e-01, 1.66313887e-01,
      9.26086754e-02, 9.67943072e-01, -2.75810599e-01,
      -6.72952756e-02, -8.33046138e-02, 1.30613551e-01,
      -2.68768333e-02, 9.00805369e-02, -1.20457843e-01,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00,
      0.00000000e+00, 0.00000000e+00, 0.00000000e+00
    ]
  ], dtype = float32),
  'center_preds': array([
    [64, 320],
    [208, 256]
  ], dtype = int64),
  'center_confs': array([
    [0.8024817],
    [0.7580417]
  ], dtype = float32),
  'cam_trans': array([
    [-1.0660633, 0.82609326, 2.563384],
    [-0.17035195, 0.69707036, 1.6288631]
  ], dtype = float32),
  'verts': array([
      [
        [2.27561016e-02, -8.49465251e-01, -2.02246144e-01],
        [1.77022722e-02, -8.34393740e-01, -2.07562864e-01],
        [2.83376612e-02, -8.32082689e-01, -1.98540911e-01],
        ...,
        [-9.59926844e-02, -8.05583000e-01, -1.08467147e-01],
        [-9.61440280e-02, -8.07182789e-01, -1.10141560e-01],
        [-9.87054482e-02, -8.04640174e-01, -1.12478524e-01]
      ],

      [
        [-5.74013777e-03, -8.62683892e-01, -2.11462319e-01],
        [-1.25920828e-02, -8.48159194e-01, -2.15562344e-01],
        [3.53422482e-04, -8.45132947e-01, -2.10286155e-01],
        ...,
        [-9.34009403e-02, -8.15017402e-01, -8.92492980e-02],
        [-9.39570889e-02, -8.16764951e-01, -9.07215476e-02],
        [-9.71432328e-02, -8.14422727e-01, -9.23290998e-02]
      ]
    ],
    dtype = float32),
  'joints': array([
      [
        [-8.75359285e-04, -2.11405411e-01, 2.78547406e-02],
        [7.71344453e-02, -1.27239078e-01, 2.42792498e-02],
        [-6.28520697e-02, -1.17797144e-01, 1.80168450e-02],
        [-1.16209695e-02, -3.09760869e-01, 6.48679286e-02],
        [1.67023391e-01, 1.37000591e-01, -2.11569801e-01],
        [-8.47365186e-02, 1.62121654e-01, -2.26580769e-01],
        [-1.47305625e-02, -4.30535108e-01, 2.35851929e-02],
        [1.64994359e-01, 4.62181956e-01, -3.93877923e-03],
        [-2.52084620e-02, 5.12286067e-01, -7.54158944e-02],
        [-1.37441671e-02, -4.67082232e-01, -1.94460638e-02],
        [2.08720669e-01, 5.55598855e-01, -8.11662003e-02],
        [-5.12957349e-02, 5.89834034e-01, -1.74600944e-01],
        [-2.08159927e-02, -6.78509831e-01, -5.24114817e-02],
        [5.85397482e-02, -5.91862917e-01, -2.98147835e-02],
        [-9.37665179e-02, -5.87976933e-01, -2.48514041e-02],
        [-1.57599710e-02, -7.27229357e-01, -1.15754671e-01],
        [1.48860753e-01, -5.89739203e-01, -4.31195199e-02],
        [-1.87773302e-01, -5.85855842e-01, -4.17570584e-02],
        [2.08096951e-01, -3.50017488e-01, 1.17042586e-02],
        [-2.41771400e-01, -3.48148525e-01, -1.39592215e-03],
        [2.16475591e-01, -1.29206508e-01, -7.70033151e-02],
        [-1.79364145e-01, -1.50368407e-01, -1.32336810e-01],
        [2.23127067e-01, -5.13947457e-02, -9.96852368e-02],
        [-1.60347641e-01, -8.30583051e-02, -1.73942208e-01],
        [-1.70510653e-02, -7.58414805e-01, -2.23913938e-01],
        [-5.17625324e-02, -7.99668312e-01, -1.99502841e-01],
        [1.39427818e-02, -8.01895857e-01, -1.98612869e-01],
        [-9.27167684e-02, -7.96374202e-01, -1.06735304e-01],
        [4.98565473e-02, -8.01094651e-01, -1.04890868e-01],
        [2.02723354e-01, 5.74146807e-01, -1.53339222e-01],
        [2.50166506e-01, 5.60621142e-01, -9.33759212e-02],
        [1.61667734e-01, 4.69619781e-01, 6.36569262e-02],
        [-2.62610093e-02, 5.97724438e-01, -2.45288610e-01],
        [-8.77840370e-02, 6.06458247e-01, -1.91111773e-01],
        [-1.72716081e-02, 5.39703488e-01, -1.13741159e-02],
        [1.51403591e-01, -5.47422171e-02, -1.65926725e-01],
        [2.09864736e-01, 1.91850066e-02, -1.69591159e-01],
        [2.36379266e-01, 3.70456576e-02, -1.51955724e-01],
        [2.49117568e-01, 3.04618478e-02, -1.22832656e-01],
        [2.55384326e-01, 1.02766156e-02, -8.32607448e-02],
        [-7.05657899e-02, -1.23868883e-01, -2.01645881e-01],
        [-1.09749392e-01, -3.75478268e-02, -2.41809100e-01],
        [-1.37292266e-01, -1.17408037e-02, -2.37244278e-01],
        [-1.61259174e-01, -6.99681044e-03, -2.15106040e-01],
        [-1.87746868e-01, -1.43653750e-02, -1.77723676e-01],
        [-1.41412631e-01, -2.21176982e-01, 3.33458707e-02],
        [1.36064485e-01, -2.43871048e-01, 3.80166210e-02],
        [-1.81482788e-02, -6.73065543e-01, -6.46214262e-02],
        [-2.47125942e-02, -9.27237451e-01, -1.38407394e-01],
        [-2.91173905e-03, -2.29785115e-01, 5.74037433e-02],
        [-1.75861567e-02, -6.15533948e-01, -4.31892462e-02],
        [-1.86750144e-02, -4.61245537e-01, 3.85091230e-02],
        [-1.10631902e-02, -7.39810348e-01, -1.47367150e-01],
        [-2.42438503e-02, -8.68159294e-01, -1.16868481e-01],
        [-5.09126931e-02, 4.62709785e-01, -6.21346831e-02],
        [-8.70343596e-02, 1.12852998e-01, -2.04238355e-01],
        [-1.54232696e-01, -2.26036042e-01, 4.65241112e-02],
        [1.47722170e-01, -2.32770562e-01, 6.69552684e-02],
        [1.73710331e-01, 1.05205201e-01, -1.99201345e-01],
        [1.68997586e-01, 4.21945602e-01, 5.61075006e-03],
        [-1.62865728e-01, -1.81941822e-01, -1.37460947e-01],
        [-2.57157743e-01, -3.64012361e-01, -3.35582905e-02],
        [-1.58776015e-01, -6.15496516e-01, -4.24131751e-02],
        [1.24996789e-01, -6.21989846e-01, -3.98044065e-02],
        [2.30993152e-01, -3.69168103e-01, -2.47584917e-02],
        [2.16143608e-01, -1.45533621e-01, -8.91659111e-02],
        [-1.66408848e-02, -6.68344438e-01, -6.19635694e-02],
        [-1.91201922e-02, -8.48284483e-01, -1.23113543e-01],
        [-2.91173905e-03, -2.29785115e-01, 5.74037433e-02],
        [-1.86750144e-02, -4.61245537e-01, 3.85091230e-02],
        [-1.10631902e-02, -7.39810348e-01, -1.47367150e-01]
      ],

      [
        [-8.72004253e-04, -2.11494520e-01, 2.78658886e-02],
        [6.69146627e-02, -1.18896343e-01, 2.28809305e-02],
        [-7.15944245e-02, -1.25474513e-01, 4.54743356e-02],
        [5.13643166e-03, -3.14690709e-01, 4.99035120e-02],
        [5.85741661e-02, 2.13484466e-01, -1.29358888e-01],
        [-1.57629505e-01, 2.08044887e-01, -9.67456549e-02],
        [3.16202501e-03, -4.34743136e-01, 6.29164651e-03],
        [5.96898869e-02, 5.26029468e-01, 9.74046588e-02],
        [-1.38327852e-01, 5.35311222e-01, 1.07657224e-01],
        [-3.62423109e-03, -4.71504658e-01, -3.60684060e-02],
        [6.61572665e-02, 6.28144503e-01, 1.89573616e-02],
        [-1.87257305e-01, 6.22663975e-01, 2.68568695e-02],
        [-9.12475213e-03, -6.83830440e-01, -6.39995337e-02],
        [6.95043057e-02, -5.93567193e-01, -6.20615482e-02],
        [-7.72789195e-02, -5.95631480e-01, -2.07347907e-02],
        [-2.07917206e-02, -7.36753404e-01, -1.23018846e-01],
        [1.51381314e-01, -5.92458904e-01, -1.02582440e-01],
        [-1.72504127e-01, -6.03741705e-01, -2.22234093e-02],
        [1.79247081e-01, -3.40870231e-01, -1.00833274e-01],
        [-2.30828285e-01, -3.63753557e-01, -3.43577750e-02],
        [9.12324637e-02, -1.72965273e-01, -2.45193064e-01],
        [-2.11503327e-01, -2.54886925e-01, -2.53490299e-01],
        [6.40000105e-02, -1.10988609e-01, -2.90337294e-01],
        [-2.05894306e-01, -2.17146173e-01, -3.25428933e-01],
        [-5.27248047e-02, -7.73335636e-01, -2.24851474e-01],
        [-7.79257268e-02, -8.13711047e-01, -1.89176619e-01],
        [-1.47682410e-02, -8.15564036e-01, -2.07549661e-01],
        [-9.00610983e-02, -8.05614114e-01, -8.89883637e-02],
        [4.71771993e-02, -8.09178174e-01, -1.28230155e-01],
        [3.98829728e-02, 6.48788810e-01, -4.76828590e-02],
        [1.02863163e-01, 6.40633702e-01, -2.60376930e-03],
        [7.30167255e-02, 5.30748844e-01, 1.64137542e-01],
        [-1.72266304e-01, 6.46206439e-01, -4.30187732e-02],
        [-2.27713630e-01, 6.35894299e-01, 1.72304213e-02],
        [-1.26233533e-01, 5.54075599e-01, 1.74781561e-01],
        [-1.58951432e-02, -1.57982886e-01, -3.21915060e-01],
        [1.05489790e-02, -7.86837339e-02, -3.68011534e-01],
        [3.22467089e-02, -4.97187972e-02, -3.66411805e-01],
        [5.30117005e-02, -4.13036644e-02, -3.43233258e-01],
        [7.53686577e-02, -4.25913036e-02, -3.04793507e-01],
        [-1.29493281e-01, -2.79582411e-01, -3.54001462e-01],
        [-1.79006279e-01, -2.14428082e-01, -4.17458266e-01],
        [-2.03316197e-01, -1.85863614e-01, -4.18327212e-01],
        [-2.19058529e-01, -1.68442532e-01, -3.95724744e-01],
        [-2.32219815e-01, -1.54547781e-01, -3.54206979e-01],
        [-1.35218203e-01, -2.37014890e-01, 5.46305366e-02],
        [1.36564359e-01, -2.26552993e-01, 9.52230417e-04],
        [-1.07802376e-02, -6.80716157e-01, -7.64743090e-02],
        [-3.14050429e-02, -9.37717974e-01, -1.32277668e-01],
        [4.66703111e-03, -2.33509585e-01, 4.95351143e-02],
        [-7.48797646e-03, -6.23024404e-01, -6.16109520e-02],
        [4.62672906e-03, -4.64726567e-01, 1.98676772e-02],
        [-2.51746681e-02, -7.50786602e-01, -1.54403657e-01],
        [-2.62231641e-02, -8.77399325e-01, -1.14986442e-01],
        [-1.53047413e-01, 4.81023699e-01, 1.14571474e-01],
        [-1.50440589e-01, 1.53840095e-01, -8.83602947e-02],
        [-1.44449502e-01, -2.46057451e-01, 6.92632571e-02],
        [1.52617037e-01, -2.19550952e-01, 2.88612153e-02],
        [7.21402764e-02, 1.78554684e-01, -1.27464011e-01],
        [7.17887357e-02, 4.86620486e-01, 1.02428593e-01],
        [-1.96798086e-01, -2.87897229e-01, -2.47597411e-01],
        [-2.51997828e-01, -3.90448928e-01, -5.74505851e-02],
        [-1.42850280e-01, -6.29573107e-01, -2.87982225e-02],
        [1.29633889e-01, -6.23592734e-01, -9.25542116e-02],
        [1.90605044e-01, -3.66491824e-01, -1.40887275e-01],
        [9.23811197e-02, -1.93608642e-01, -2.50356674e-01],
        [-7.82386772e-03, -6.77114964e-01, -7.55808204e-02],
        [-2.35089418e-02, -8.58024240e-01, -1.23406678e-01],
        [4.66703111e-03, -2.33509585e-01, 4.95351143e-02],
        [4.62672906e-03, -4.64726567e-01, 1.98676772e-02],
        [-2.51746681e-02, -7.50786602e-01, -1.54403657e-01]
      ]
    ],
    dtype = float32),
  'pj2d_org': array([
    [
      [278.6843, 949.2278],
      [329.72742, 1004.2992],
      [238.13193, 1010.4773],
      [271.65326, 884.8723],
      [388.5432, 1177.1956],
      [223.81256, 1193.6327],
      [269.61862, 805.84766],
      [387.2156, 1389.9672],
      [262.76276, 1422.751],
      [270.264, 781.9343],
      [415.82645, 1451.0913],
      [245.69344, 1473.492],
      [265.63684, 643.59375],
      [317.5606, 700.2882],
      [217.90408, 702.83093],
      [268.94507, 611.7158],
      [376.65912, 701.67773],
      [156.39392, 704.2189],
      [415.4183, 858.5316],
      [121.06207, 859.7545],
      [420.9006, 1003.01196],
      [161.8962, 989.1653],
      [425.25278, 1053.9254],
      [174.33896, 1033.2074],
      [268.10028, 591.3106],
      [245.388, 564.31775],
      [288.38007, 562.8602],
      [218.59097, 566.47314],
      [311.87903, 563.3844],
      [411.90228, 1463.2275],
      [442.94513, 1454.3774],
      [385.03894, 1394.8339],
      [262.07404, 1478.6547],
      [221.81854, 1484.3694],
      [267.95596, 1440.6907],
      [378.3229, 1051.7351],
      [416.575, 1100.1069],
      [433.92392, 1111.7935],
      [442.25876, 1107.4855],
      [446.35922, 1094.278],
      [233.08473, 1006.5044],
      [207.44626, 1062.9856],
      [189.4245, 1079.8716],
      [173.74254, 1082.9757],
      [156.41121, 1078.1543],
      [186.72844, 942.8341],
      [368.28632, 927.985],
      [267.38232, 647.156],
      [263.0872, 480.84717],
      [277.35187, 937.20166],
      [267.75015, 684.7999],
      [267.0377, 785.7533],
      [272.01822, 603.4838],
      [263.3939, 519.503],
      [245.94406, 1390.3125],
      [222.30904, 1161.3954],
      [178.34009, 939.65466],
      [375.9141, 935.2483],
      [392.91858, 1156.3914],
      [389.83496, 1363.6398],
      [172.69135, 968.50635],
      [110.99453, 849.3745],
      [175.36731, 684.82446],
      [361.0445, 680.5757],
      [430.39972, 846.0011],
      [420.68338, 992.32886],
      [268.36865, 650.2451],
      [266.7464, 532.5074],
      [277.35187, 937.20166],
      [267.0377, 785.7533],
      [272.01822, 603.4838]
    ],

    [
      [783.03174, 1054.7114],
      [856.94324, 1155.6763],
      [705.91925, 1148.5037],
      [789.58307, 942.1909],
      [847.8491, 1518.089],
      [612.11053, 1512.158],
      [787.43024, 811.2914],
      [849.0657, 1858.874],
      [633.1562, 1868.9941],
      [780.0308, 771.20825],
      [856.11743, 1970.2153],
      [579.8058, 1964.2397],
      [774.0333, 539.69794],
      [859.76685, 638.1168],
      [699.7211, 635.86597],
      [761.3122, 481.99316],
      [949.0418, 639.3253],
      [595.8919, 627.02295],
      [979.4253, 913.646],
      [532.2979, 888.69507],
      [883.45825, 1096.7219],
      [553.369, 1007.3983],
      [853.7652, 1164.2985],
      [559.4848, 1048.5491],
      [726.49384, 442.10553],
      [699.01587, 398.08203],
      [767.8799, 396.06165],
      [685.784, 406.91058],
      [835.42236, 403.02448],
      [827.46906, 1992.7249],
      [896.1398, 1983.833],
      [863.5966, 1864.0195],
      [596.15125, 1989.9092],
      [535.694, 1978.6653],
      [646.34326, 1889.4541],
      [766.6512, 1113.0581],
      [795.4847, 1199.5222],
      [819.1429, 1231.1042],
      [841.78406, 1240.2798],
      [866.1611, 1238.8759],
      [642.789, 980.47144],
      [588.8023, 1051.5128],
      [562.29584, 1082.6582],
      [545.1311, 1101.6533],
      [530.78064, 1116.8035],
      [636.54675, 1026.8853],
      [932.8861, 1038.2924],
      [772.2283, 543.09357],
      [749.73987, 262.87054],
      [789.0713, 1030.7072],
      [775.818, 605.99805],
      [789.02734, 778.59875],
      [756.5332, 466.69202],
      [755.39, 328.63922],
      [617.1066, 1809.8018],
      [619.949, 1453.0555],
      [626.4814, 1017.0255],
      [950.38916, 1045.927],
      [862.641, 1480.0033],
      [862.2577, 1815.904],
      [569.4029, 971.4054],
      [509.21564, 859.58765],
      [628.22516, 598.85767],
      [925.3294, 605.3784],
      [991.8096, 885.7095],
      [884.7107, 1074.2134],
      [775.4518, 547.02014],
      [758.3494, 349.76495],
      [789.0713, 1030.7072],
      [789.02734, 778.59875],
      [756.5332, 466.69202]
    ]
  ], dtype = float32)
}

Exported according to the README by:

import numpy as np
results = np.load('E:/Projekte/romp-test/00000215.npz', allow_pickle=True)['results'][()]
print(results)

My pip freeze output:

black==22.3.0
certifi==2021.10.8
click==8.1.2
colorama==0.4.4
Cython==0.29.30
lap==0.4.0
mypy-extensions==0.4.3
numpy==1.22.3
opencv-python==4.5.5.64
pathspec==0.9.0
platformdirs==2.5.1
simple-romp==1.0.5
tomli==2.0.1
torch==1.11.0
typing_extensions==4.1.1
wincertstore==0.2

System:

  • Python: v3.9.12
  • OS: Windows 10 Enterprise (64-bit, Version 21H2, Build 19044.1706)

vivi90 avatar Jun 09 '22 00:06 vivi90

Hi, Vivian,

Thanks for reporting this bug. I will take care of it.

Arthur151 avatar Jun 21 '22 14:06 Arthur151