linghu8812 / tensorrt_inference

705 stars 207 forks source link

img_mean for nanodet #27

Closed thunder95 closed 3 years ago

thunder95 commented 3 years ago

Hello, in you code of naodet, the img_std is exactly the same with img_mean, any reason for that? or its a bug there?

    img_mean = config["img_mean"].as<std::vector<float>>();
    img_std = config["img_mean"].as<std::vector<float>>();
linghu8812 commented 3 years ago

@thunder95 sorry, it's a mistake, it should be:

img_mean = config["img_mean"].as<std::vector<float>>();
img_std = config["img_std "].as<std::vector<float>>();
thunder95 commented 3 years ago

ok, thx for your feedback