-
Notifications
You must be signed in to change notification settings - Fork 3
/
config.ini
89 lines (82 loc) · 3.65 KB
/
config.ini
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
[General] # This section contains general config
# Flag to indicate if using camera or video files
use_camera = False
# Camera id for query camera, for use_camera=True
query_cam = 0
# Camera ids for gallery cameras, for use_camera=True (eg. 1,2,3)
gallery_cams = 0
# Path to the query video, for use_camera=False
query_video = Test_Videos\MVI_39401.mp4
# Path to the folder containing all gallery videos, for use_camera=False
gallery_video_folder = Test_Videos\DT
# Wait key delay, for use_camera=False
wait_delay = 30
# Flag to save video
save_video = False
# Saved video name
save_video_name = ReID_Compare.mp4
# Folder path to save video
save_video_path = output
# Print verbose message
verbose = True
# Flag to Disable CUDA, use CPU. Flag not in use if cuda not available
disable_cuda = False
# Flag to indicate if the processed frames should be displayed
display_frames = True
# ---------- Algorithm config ----------
# Detector or Tracker algorithm, need to match dictonary string in algorithm factory and section in config.ini
# eg (Choose one): [YOLOv3, JDE, NoDetector]
# Choose NoDetector to indicate no detection or tracking algorithm is to be used
det_algorithm = JDE
# ReID algorithm, need to match dictonary string in algorithm factory and section in config.ini
# eg (Choose one): [PersonReID, DeepPersonReID, NoReID]
# Choose NoReID to indicate no ReID algorithm is to be used, automatically set to NoReID if no valid detector or tracker algorithm is set
reid_algorithm = PersonReID
# ------------------------------
# ---------- Detector config ----------
# Detector type only requires one instance of the algorithm to perform inference across different cameras/videos
[NoDetector] # Placeholder for no detector selected
type = detector
[YOLOv3]
# YOLOv3 model configuration (For tiny-yolo, replace model_config and weights)
type = detector
model_config = Detector_Tracker/YOLOv3/cfg/yolov3.cfg
weights = Detector_Tracker/YOLOv3/weights/yolov3.weights
labels = Detector_Tracker/YOLOv3/data/coco.names
confidence_threshold = 0.5
score_threshold = 0.5
iou_threshold = 0.5
# This is to filter the detection for only the classes required, see labels file for index (e.g. for coco index 2 = car)
class_filter = 2
# ------------------------------
# ---------- Tracker config ----------
# Tracker type requires one instance of the algorithm to perform inference for each different cameras/videos
# Algorithm class instances will be stored as a list in MultiVideoProcessor class
[JDE] # https://github.com/Zhongdao/Towards-Realtime-MOT
type = tracker
network_config = Detector_Tracker/JDE/cfg/yolov3_864x480_detrac.cfg
weights = Detector_Tracker/JDE/weights/final_1st.pt
iou_threshold = 0.5
conf_threshold = 0.5
nms_threshold = 0.4
min_box_area = 200
track_buffer = 30
frame_rate = 30
# ------------------------------
# ---------- ReID config ----------
[NoReID] # Placeholder for no ReID selected
type = ReID
[PersonReID] # https://github.com/layumi/Person_reID_baseline_pytorch
# PersonReID model configuration
type = ReID
network_config = ReID/PersonReID/detrac4_ResNet50/opts.yaml
weights = ReID/PersonReID/detrac4_ResNet50/net_last.pth
match_threshold = 0.6
[DeepPersonReID] #https://github.com/KaiyangZhou/deep-person-reid
# NOTE: To use DeepPersonReID, additional setup is required, see 'installation' section in Github link above
# NOTE: For model naming, see Github link
type = ReID
model = osnet_ain_x1_0
weights = ReID/DeepPersonReID/model/osnet_ain_x1_0_mars_softmax_cosinelr/model.pth.tar-150
match_threshold = 0.6
# ------------------------------