isarandi / metrabs

Estimate absolute 3D human poses from RGB images.

Home Page:https://arxiv.org/abs/2007.07227

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

running inference with `max_detections` parameter results in failure

vladmandic opened this issue · comments

on some inputs (not all), using max_detections parameter results in failure:
(re-running inference without max_detections specified works just fine)

using parameters:

  model:metrabs_eff2l_y4_360
  fov:55
  batch:64
  maxpeople:4
  skeleton:smpl+head_30
  augmentations:6
  average:1
  suppress:1
  minconfidence:0.1
  iou:0.7

actual inference call:

  result = model.detect_poses(tensor,
    default_fov_degrees=args.fov,
    internal_batch_size=args.batch,
    num_aug=args.augmentations,
    average_aug=bool(args.average),
    skeleton=args.skeleton,
    detector_threshold=args.minconfidence,
    detector_nms_iou_threshold=args.iou,
    max_detections=args.maxpeople,
    antialias_factor=1,
    suppress_implausible_poses=bool(args.suppress)
  )

error log

  File "/home/vlado/.local/lib/python3.9/site-packages/tensorflow/python/util/traceback_utils.py", line 153, in error_handler
    raise e.with_traceback(filtered_tb) from None
  File "/home/vlado/.local/lib/python3.9/site-packages/tensorflow/python/eager/execute.py", line 54, in quick_execute
    tensors = pywrap_tfe.TFE_Py_Execute(ctx._handle, device_name, op_name,

tensorflow.python.framework.errors_impl.InvalidArgumentError: Graph execution error:

Detected at node 'cond_2/TopKV2' defined at (most recent call last):
    File "/home/vlado/dev/motioniq/process/process.py", line 51, in loadModel
      model = tf.saved_model.load(args.model)
Node: 'cond_2/TopKV2'
1 root error(s) found.
  (0) INVALID_ARGUMENT:  input must have at least k columns. Had 3, needed 4
         [[{{node cond_2/TopKV2}}]]
         [[StatefulPartitionedCall/StatefulPartitionedCall/cond_3/else/_4381/cond_3/cond_7/then/_11169/cond_3/cond_7/map/while/loop_body_control/_18889/_1451]]
0 successful operations.
0 derived errors ignored. [Op:__inference_restored_function_body_457118]

p.s. i love your models - extremely well made!

This should be fixed in the new models listed at https://istvansarandi.com/dozens