-
Notifications
You must be signed in to change notification settings - Fork 13
/
train.sh
executable file
·117 lines (106 loc) · 4.91 KB
/
train.sh
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
cd ../../../..
# # # # # # # # # # # -----------------------------------------------------------
# # ## uncomment for hrnet w48-384
# # # # # # # # # # -----------------------------------------------------------
CUDA_VISIBLE_DEVICES=0,1,2, python tools/train.py \
--cfg experiments/crowdpose/hrnet/w48_384x288_adam_lr1e-3.yaml \
GPUS '(0,1,2)' \
OUTPUT_DIR 'Outputs/outputs/crowdpose'\
LOG_DIR 'Outputs/logs/crowdpose'\
DATASET.DATASET 'crowdpose' \
DATASET.TRAIN_DATASET 'crowdpose' \
DATASET.TRAIN_SET 'train' \
DATASET.TRAIN_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
DATASET.TRAIN_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_train.json' \
DATASET.TRAIN_DATASET_TYPE 'crowdpose' \
DATASET.TEST_DATASET 'crowdpose' \
DATASET.TEST_SET 'val' \
DATASET.TEST_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
DATASET.TEST_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_val.json' \
DATASET.TEST_DATASET_TYPE 'crowdpose' \
TRAIN.LR 0.001 \
TRAIN.END_EPOCH 210 \
TRAIN.BATCH_SIZE_PER_GPU 10 \
TEST.BATCH_SIZE_PER_GPU 256 \
TEST.USE_GT_BBOX True \
EPOCH_EVAL_FREQ 1 \
PRINT_FREQ 100 \
MODEL.NAME 'pose_hrnet' \
# # ## uncomment for hrnet w48-256
# # # # # # # # # # # -----------------------------------------------------------
# CUDA_VISIBLE_DEVICES=0,1,2, python tools/train.py \
# --cfg experiments/crowdpose/hrnet/w48_256x192_adam_lr1e-3.yaml \
# GPUS '(0,1,2)' \
# OUTPUT_DIR 'Outputs/outputs/crowdpose'\
# LOG_DIR 'Outputs/logs/crowdpose'\
# DATASET.DATASET 'crowdpose' \
# DATASET.TRAIN_DATASET 'crowdpose' \
# DATASET.TRAIN_SET 'train' \
# DATASET.TRAIN_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
# DATASET.TRAIN_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_train.json' \
# DATASET.TRAIN_DATASET_TYPE 'crowdpose' \
# DATASET.TEST_DATASET 'crowdpose' \
# DATASET.TEST_SET 'val' \
# DATASET.TEST_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
# DATASET.TEST_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_val.json' \
# DATASET.TEST_DATASET_TYPE 'crowdpose' \
# TRAIN.LR 0.001 \
# TRAIN.END_EPOCH 210 \
# TRAIN.BATCH_SIZE_PER_GPU 10 \
# TEST.BATCH_SIZE_PER_GPU 256 \
# TEST.USE_GT_BBOX True \
# EPOCH_EVAL_FREQ 1 \
# PRINT_FREQ 100 \
# MODEL.NAME 'pose_hrnet' \
# # ## uncomment for hrnet w32-384
# # # # # # # # # # # -----------------------------------------------------------
# CUDA_VISIBLE_DEVICES=0,1,2, python tools/train.py \
# --cfg experiments/crowdpose/hrnet/w32_384x288_adam_lr1e-3.yaml \
# GPUS '(0,1,2)' \
# OUTPUT_DIR 'Outputs/outputs/crowdpose'\
# LOG_DIR 'Outputs/logs/crowdpose'\
# DATASET.DATASET 'crowdpose' \
# DATASET.TRAIN_DATASET 'crowdpose' \
# DATASET.TRAIN_SET 'train' \
# DATASET.TRAIN_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
# DATASET.TRAIN_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_train.json' \
# DATASET.TRAIN_DATASET_TYPE 'crowdpose' \
# DATASET.TEST_DATASET 'crowdpose' \
# DATASET.TEST_SET 'val' \
# DATASET.TEST_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
# DATASET.TEST_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_val.json' \
# DATASET.TEST_DATASET_TYPE 'crowdpose' \
# TRAIN.LR 0.001 \
# TRAIN.END_EPOCH 210 \
# TRAIN.BATCH_SIZE_PER_GPU 64 \
# TEST.BATCH_SIZE_PER_GPU 256 \
# TEST.USE_GT_BBOX True \
# EPOCH_EVAL_FREQ 1 \
# PRINT_FREQ 100 \
# MODEL.NAME 'pose_hrnet' \
# # ## uncomment for hrnet w32-256
# # # # # # # # # # # -----------------------------------------------------------
# CUDA_VISIBLE_DEVICES=0,1,2, python tools/train.py \
# --cfg experiments/crowdpose/hrnet/w32_256x192_adam_lr1e-3.yaml \
# GPUS '(0,1,2)' \
# OUTPUT_DIR 'Outputs/outputs/crowdpose'\
# LOG_DIR 'Outputs/logs/crowdpose'\
# DATASET.DATASET 'crowdpose' \
# DATASET.TRAIN_DATASET 'crowdpose' \
# DATASET.TRAIN_SET 'train' \
# DATASET.TRAIN_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
# DATASET.TRAIN_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_train.json' \
# DATASET.TRAIN_DATASET_TYPE 'crowdpose' \
# DATASET.TEST_DATASET 'crowdpose' \
# DATASET.TEST_SET 'val' \
# DATASET.TEST_IMAGE_DIR '/home/XYZ/Desktop/datasets/crowdpose/images'\
# DATASET.TEST_ANNOTATION_FILE '/home/XYZ/Desktop/datasets/crowdpose/json/crowdpose_val.json' \
# DATASET.TEST_DATASET_TYPE 'crowdpose' \
# TRAIN.LR 0.001 \
# TRAIN.END_EPOCH 210 \
# TRAIN.BATCH_SIZE_PER_GPU 64 \
# TEST.BATCH_SIZE_PER_GPU 256 \
# TEST.USE_GT_BBOX True \
# EPOCH_EVAL_FREQ 1 \
# PRINT_FREQ 100 \
# MODEL.NAME 'pose_hrnet' \