Retraining BodyPoseNet

Accelerated Computing Intelligent Video Analytics TAO Toolkit
4

Refer to BodyPoseNet TAO training error - #4 by Morganh

  • The target_shape depends on the input shape. This can be computed based on the model stride. In the default setting, the model has a stride of 8.

The assertion error is due to

assert (image_shape[0] // target_shape[0]) == (image_shape[1] // target_shape[1])