-
Notifications
You must be signed in to change notification settings - Fork 16
/
get_opts.m
95 lines (85 loc) · 2.73 KB
/
get_opts.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
function opts = get_opts()
addpath(genpath('src'))
opts = [];
opts.dataset = 0;%0 for duke, 1 for mot, 2 for aic
opts.feature_dir = [];
opts.dataset_path = 'D:/Data/DukeMTMC/';
opts.gurobi_path = 'C:/Utils/gurobi801/win64/matlab';
opts.experiment_root = 'experiments';
opts.experiment_name = 'demo';
opts.reader = MyVideoReader(opts.dataset_path);
% General settings
opts.eval_dir = 'L3-identities';
opts.visualize = false;
opts.image_width = 1920;
opts.image_height = 1080;
opts.current_camera = -1;
opts.world = 0;
opts.ROIs = getROIs();
opts.minimum_trajectory_length = 100;
opts.optimization = 'BIPCC';
opts.num_cam = 8;
opts.sequence = 2;
opts.sequence_names = {'trainval', 'trainval_mini', 'test_easy', 'test_hard', 'trainval_nano','test_all','train','val'};
opts.sequence_intervals = {47720:227540, 127720:187540, 263504:356648, 227541:263503, 127720:127840,227541:356648,47720:187540,187541:227540};
opts.start_frames = [5543, 3607, 27244, 31182, 1, 22402, 18968, 46766];
opts.render_threshold = 0.05;
opts.load_tracklets = 1;
opts.load_trajectories = 1;
opts.appear_model_name = '1fps_train_IDE_40/model_param_L2_75.mat';
opts.motion_model_name = '1fps_train_IDE_40/model_param_L2_motion_150.mat';
opts.fft = false;
opts.soft = 0.1;
opts.fft = 60000/1001;
% Tracklets
tracklets = [];
tracklets.spatial_groups = 1;
tracklets.window_width = 50;
tracklets.min_length = 5;
tracklets.alpha = 1;
tracklets.beta = 0.02;
tracklets.cluster_coeff = 0.75;
tracklets.nearest_neighbors = 8;
tracklets.speed_limit = 20;
tracklets.threshold = 8;
tracklets.diff_p = 0;
tracklets.diff_n = 0;
tracklets.step = false;
tracklets.og_appear_score = true;
tracklets.og_motion_score = true;
% Trajectories
trajectories = [];
trajectories.appearance_groups = 0; % determined automatically when zero
trajectories.alpha = 1;
trajectories.beta = 0.01;
trajectories.window_width = 300;
trajectories.speed_limit = 30;
trajectories.indifference_time = 100;
trajectories.threshold = 8;
trajectories.diff_p = 0;
trajectories.diff_n = 0;
trajectories.step = false;
trajectories.og_appear_score = true;
trajectories.og_motion_score = true;
trajectories.use_indiff = true;
% Identities
identities = [];
identities.window_width = 5000;
identities.appearance_groups = 0; % determined automatically when zero
identities.alpha = 1;
identities.beta = 0.01;
identities.speed_limit = 2.5;
identities.indifference_time = 6000;
identities.threshold = 8;
identities.diff_p = 0;
identities.diff_n = 0;
identities.optimal_filter = true;
identities.step = false;
identities.extract_images = true;
identities.og_appear_score = true;
identities.og_motion_score = true;
identities.reintro_time_matrix = ones(1,8)*inf;
opts.tracklets = tracklets;
opts.trajectories = trajectories;
opts.identities = identities;
end