linghu8812 / tensorrt_inference

702 stars 206 forks source link

The code use include "classification.h" instead of #include "model.h" #177

Open copilot0058 opened 1 day ago

copilot0058 commented 1 day ago

The code use #include "classification.h" instead of #include "model.h"

Anyone know which is the better ? case1 . Using #include "classification.h" on the detection.h case1 . Using #include "model.h" on the detection.h

https://github.com/linghu8812/tensorrt_inference/blob/8d115d9806d9e09dc260f9fb6cf05297f4968d6e/code/includes/detection.h#L4C1-L5C1

copilot0058 commented 1 day ago

And I found the main include chain

main.cpp -> build.h -> yolov8.h -> detection.h -> classification.h -> model.h -> common.h

And it only use build_model function on main.cpp. Where the build_model function use the class model in model.h

why directly include model.h?

https://github.com/linghu8812/tensorrt_inference/blob/master/project/main.cpp