search

RangiLyu / nanodet

NanoDet-Plus⚡Super fast and lightweight anchor-free object detection model. 🔥Only 980 KB(int8) / 1.8MB (fp16) and run 97FPS on cellphone🔥
Apache License 2.0
5.63k stars 1.03k forks source link

请问怎样可以将demo.py最终输出的bounding box输出？ #567

Closed ChenJiajun0011 closed 1 month ago

ChenJiajun0011 commented 1 month ago

运行demo.py后输出的bounding box是在哪里输出的呢？我看到visualize函数里面是画框，但是是把不同置信度的框画出来，没有看到哪里调用了NMS去输出最终的box啊，求教是在哪里输出的！球球了

Sanath1998 commented 1 month ago

Hi @ChenJiajun0011, NMS function is being called in "def get_bboxes(self, cls_preds, reg_preds, img_metas)" api. This can be seen inside /nanodet/model/head/nanodet_plus_head.py script.

ChenJiajun0011 commented 1 month ago

Hi @ChenJiajun0011, NMS function is being called in "def get_bboxes(self, cls_preds, reg_preds, img_metas)" api. This can be seen inside /nanodet/model/head/nanodet_plus_head.py script.

Okey got you, thanks!