From fe34611ad11b07b4bcb97778fcb08ecae0886f59 Mon Sep 17 00:00:00 2001 From: lzc6996 Date: Sat, 31 Jul 2021 17:14:45 +0800 Subject: [PATCH] convert to relative path import --- sot_3d/tracker.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/sot_3d/tracker.py b/sot_3d/tracker.py index 03983ad..237fe72 100644 --- a/sot_3d/tracker.py +++ b/sot_3d/tracker.py @@ -3,8 +3,13 @@ import os, numpy as np from scipy.optimize import minimize import matplotlib.pyplot as plt -from . import OptimData, FrameData, DataBuffer, LossFunc, MotionModel, ShapeMap, finder +from . import FrameData, OptimData +from .loss_func import LossFunc +from .motion_model import MotionModel +from .shape_map import ShapeMap from .data_protos import BBox +from .data_buffer import DataBuffer +from .finder import Finder class Tracker: @@ -103,7 +108,7 @@ def pre_frame_optim(self, input_data: FrameData): elif self.cur_frame == 1: self.tmp_init_pc1 = input_data.pc - finder = finder.Finder(self.configs['finder'], self.tmp_init_box0, self.tmp_init_pc0, self.tmp_init_pc1) + finder = Finder(self.configs['finder'], self.tmp_init_box0, self.tmp_init_pc0, self.tmp_init_pc1) pred_motion = finder.icp() del self.tmp_init_pc0, self.tmp_init_pc1 else: