CenterTrack
dandeseed
知足且上进,温柔且坚定
展开
-
Tracker类
class Tracker(object): def __init__(self, opt): self.opt = opt self.reset() def init_track(self, results): for item in results: if item['score'] > self.opt.new_thresh: self.id_count += 1 # active and age are never原创 2020-05-28 11:45:07 · 505 阅读 · 0 评论 -
Demo推断阶段流程图
Created with Raphaël 2.2.0初始化Detector类While True进入detector.run进入detector.process()进入detector.model()跳出detector.model()跳出detector.process()进入detector.post_process()跳出detector.post_process()进入detector.tracker.step()跳出detector.tracker.step()进入detector.show_re原创 2020-05-28 11:44:22 · 483 阅读 · 1 评论 -
debugger.py
class Debugger(object): def __init__(self, opt, dataset): self.opt = opt self.imgs = {} self.theme = opt.debugger_theme self.plt = plt self.with_3d = False self.names = dataset.class_name self.out_size = 384 if opt.dataset ==原创 2020-05-28 09:52:58 · 289 阅读 · 0 评论 -
demo函数
def demo(opt): os.environ['CUDA_VISIBLE_DEVICES'] = opt.gpus_str # 指定gpu的id号 opt.debug = max(opt.debug, 1) # 选择可视化的等级1,2,3,4 detector = Detector(opt) # 检测器对象 if opt.demo == 'webcam' or \ # 判断输入的是video原创 2020-05-27 17:58:22 · 3935 阅读 · 0 评论 -
Opts类
class opts(object): def __init__(self): self.parser = argparse.ArgumentParser() # basic experiment setting self.parser.add_argument('task', default='', help='ctdet | ddd | multi_pose '原创 2020-05-27 15:22:13 · 758 阅读 · 0 评论 -
CenterTrack代码inference阶段框架
demo.pyOpts对象:包含所有的超参数class opts(objects): def __init__(self): def parse(self,args=''): return opt def update_dataset_info_and_set_heads(self,opt,dataset): return opt def init(self,args=''): return optDetector对象原创 2020-05-27 15:14:19 · 1407 阅读 · 5 评论 -
Detector(object)
class Detector(object): def __init__(self, opt): if opt.gpus[0] >= 0: opt.device = torch.device('cuda') else: opt.device = torch.device('cpu') print('Creating model...') self.model = create_model( opt.arch, opt.h原创 2020-05-27 14:47:06 · 1085 阅读 · 5 评论