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