different image shape about isbfsar

[steb6]

Hi, did you rename hpe.py to demo.py?
Try to clone my latest push and run modules/hpe/hpe.py.
The system works both with rgb or skeleton input data and the input data type can be changed in utils/params/py.
Since this is a work in development some scripts may not work anymore, you may want to check main.py that is the most updated one.

[luoww1992]

Hi, did you rename hpe.py to demo.py? Try to clone my latest push and run modules/hpe/hpe.py. The system works both with rgb or skeleton input data and the input data type can be changed in utils/params/py. Since this is a work in development some scripts may not work anymore, you may want to check main.py that is the most updated one.

yes， i copy the file hpe.py and rename it with demo.py to edit it , i am chenking the code,
i find that maybe i need to change the batch_size=5 to batch_size=1, i think it is the error place, now i am trying to run the setup again.

[steb6]

You are right, I have created 2 TRT engines: one with batch_size=1 and one with batch_size=5 for test time augmentation.