Extract features for object tracking

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors

GNU General Public License v3.0

13.05k stars 4.13k forks source link

Extract features for object tracking #1302

Open danwanban opened 1 year ago

danwanban commented 1 year ago

Hi,

Is it possible to take the output of intermediate layer/s before nms for features to use later in a tracking module ? If so - which layer/s should I use ?

Thanks for the help !

yulin010101 commented 1 year ago

NMS is post-processing and is not included in the model architecture.