search

zengwb-lx / Yolov5-Deepsort-Fastreid

GNU General Public License v3.0
274 stars 63 forks source link

想用Reid_feature替换原来的extractor #12

Open Edward-YS opened 2 months ago

Edward-YS commented 2 months ago

作者你好！ 我想把deep_sort.py中的extractor替换成Reid_feature，抱歉我是一个小白，没有看懂你写的，这个提取器是FeatureExtractionDemo里面的吗？ 还望指教！谢谢！

zengwb-lx commented 2 months ago

用Reid_feature替换原来的extractor，我有写好的，直接替换模型就行

Edward-YS commented 2 months ago

大佬 具体是在哪里啊？ 我看Reid_feature是一个类 class Reid_feature(): def init(self): args = get_parser().parse_args() cfg = setup_cfg(args) self.demo = FeatureExtractionDemo(cfg, parallel=args.parallel)

def __call__(self, img_list):
    import time
    t1 = time.time()
    feat = self.demo.run_on_image(img_list)
    # print('reid time:', time.time() - t1, len(img_list))
    return feat
   里面又有一个FeatureExtractionDemo，有点被绕晕了