-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathconfig.py
49 lines (43 loc) · 2.33 KB
/
config.py
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
# SPDX-FileCopyrightText: 2025 Idiap Research Institute <[email protected]>
# SPDX-FileContributor: Anshul Gupta <[email protected]>
#
# SPDX-License-Identifier: GPL-3.0-only
# =============================================================================
# GazeFollow dataset dir config
# =============================================================================
gazefollow_data = ".../gazefollow_extended"
gazefollow_depth = ".../gazefollow/depth_geom"
gazefollow_focal_length = ".../gazefollow/focal_length"
gazefollow_train_label = ".../labels/gazefollow/train_annotations_new.txt"
gazefollow_val_label = ".../labels/gazefollow/val_annotations_new.txt"
gazefollow_val_crop_params = ".../val_crop_params/gazefollow_val_crop_params.txt"
gazefollow_test_label = ".../labels/gazefollow/test_annotations_release.txt"
# =============================================================================
# VideoAttTarget dataset dir config
# =============================================================================
videoattentiontarget_data = ".../VideoAttentionTarget/images"
videoattentiontarget_depth = ".../VideoAttentionTarget/depth_geom/images"
videoattentiontarget_focal_length = ".../VideoAttentionTarget/focal_length"
videoattentiontarget_train_label = ".../labels/vat/train_annotations.txt"
videoattentiontarget_val_label = ".../labels/vat/val_annotations.txt"
videoattentiontarget_val_crop_params = ".../val_crop_params/vat_val_crop_params.txt"
videoattentiontarget_test_label = ".../labels/vat/test_annotations_pid.txt"
# =============================================================================
# ChildPlay dataset dir config
# =============================================================================
childplay_data = ".../ChildPlay"
childplay_label = '.../ChildPlay'
childplay_depth = ".../ChildPlay/depth"
childplay_focal_length = ".../ChildPlay/focal_length"
childplay_val_crop_params = ".../val_crop_params/childplay_val_crop_params.txt"
# human-centric branch weights
human_centric_weights = '.../human_centric.pt'
# =============================================================================
# model config
# =============================================================================
input_size_scene = (512, 288)
input_size_human = (224, 224)
output_size = (128, 72)
use_amp = False
perform_swa = True
sched_type = 'step' # {onecycle, step}