Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

mot_eval #9191

Open
wants to merge 2 commits into
base: develop
Choose a base branch
from
Open

mot_eval #9191

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
2 commits
Select commit Hold shift + click to select a range
3ccdbfc
mot_eval
lunaticsst Oct 24, 2024
b10d21f
mot_eval_fix
lunaticsst Oct 24, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions configs/mot/fairmot/fairmot_dla34_30e_1088x608.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,3 +7,10 @@ _BASE_: [
]

weights: output/fairmot_dla34_30e_1088x608/model_final

EvalDataset:
name: MOTImageFolder
dataset_dir: dataset/mot
data_root: MOT17/images/train
keep_ori_im: False # set True if save visualization images or video, or used in DeepSORT
data_fields: ['image']
2 changes: 2 additions & 0 deletions ppdet/engine/callbacks.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -213,6 +213,8 @@ def on_epoch_end(self, status):
key = 'bbox'
elif 'keypoint' in map_res:
key = 'keypoint'
elif 'MOTA' in map_res:
key = 'MOTA'
else:
key = 'mask'

Expand Down
15 changes: 9 additions & 6 deletions ppdet/engine/tracker.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -128,16 +128,19 @@ def register_metrics(self, metrics):
def load_weights_jde(self, weights):
load_weight(self.model, weights, self.optimizer)

def load_weights_sde(self, det_weights, reid_weights):
def load_weights_sde(self, det_weights, reid_weights, training_weights=False, weights=None):
with_detector = self.model.detector is not None
with_reid = self.model.reid is not None

if with_detector:
load_weight(self.model.detector, det_weights)
if with_reid:
load_weight(self.model.reid, reid_weights)
if training_weights:
load_weight(self.model, weights)
else:
load_weight(self.model.reid, reid_weights)
if with_detector:
load_weight(self.model.detector, det_weights)
if with_reid:
load_weight(self.model.reid, reid_weights)
else:
load_weight(self.model.reid, reid_weights)

def _eval_seq_centertrack(self,
dataloader,
Expand Down
Loading