History

haotian 7ac785ddb3 first commit		2025-03-07 11:35:40 +08:00
..
common.hpp	first commit	2025-03-07 11:35:40 +08:00
demo.py	first commit	2025-03-07 11:35:40 +08:00
hrnet.cpp	first commit	2025-03-07 11:35:40 +08:00
logging.h	first commit	2025-03-07 11:35:40 +08:00
README.md	first commit	2025-03-07 11:35:40 +08:00

README.md

HRNet

The Pytorch implementation is HRNet-Image-Classification. The implemented model is HRNet-W18-C-Small-v2

How to Run

1. generate .wts
Download code and model from HRNet-Image-Classification and config your environments.

Put demo.py in the YOUR_ROOT_DIR\HRNet-Image-Classification\tools folder, set savewts in main() as True, and run, the .wts will be generated.

cmake and make

mkdir build
cd build
cmake ..
make
sudo ./hrnet -s             // serialize model to plan file i.e. 'hrnet.engine'
sudo ./hrnet -d  ../samples // deserialize plan file and run inference, the images in samples will be processed.

Result

The test img:

Pytorch Result:

TRT Result: