-
Notifications
You must be signed in to change notification settings - Fork 5
/
run.sh
90 lines (85 loc) · 3.32 KB
/
run.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
# ##### emotic training scripts #####
# YOUR_DATA_PATH=/home/lxp/data/emotic
# CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch \
# --nproc_per_node=1 \
# --master_port=29507 \
# --use_env main.py \
# --dataset_file=emotic \
# --backbone=resnet50 \
# --binary_flag=1 \
# --detr=deformable_detr_dsct \
# --model=deformable_transformer_dsct \
# --batch_size=4 \
# --cls_loss_coef=5 \
# --data_path=$YOUR_DATA_PATH \
# --output_dir=$YOUR_DATA_PATH/checkpoints \
# --epochs=50 \
# --lr_drop=40 \
# --num_queries=4 \
# --pretrained_weights=./r50_deformable_detr-checkpoint.pth \
# ##### emotic testing scripts #####
# YOUR_DATA_PATH=/home/lxp/data/emotic
# YOUR_MODEL_PATH=/home/lxp/data/emotic/checkpoints
# CUDA_VISIBLE_DEVICES=0 python test.py \
# --dataset_file=emotic \
# --detr=deformable_detr_dsct \
# --model=deformable_transformer_dsct \
# --num_queries=4 \
# --binary_flag=1 \
# --data_path=$YOUR_DATA_PATH/images \
# --json_path=./datasets/annotations/emotic_test_bi.json \
# --pretrained_weights=$YOUR_MODEL_PATH/checkpoint.pth;
# ##### emotic visualization scripts #####
# YOUR_DATA_PATH=/home/lxp/data/emotic
# YOUR_MODEL_PATH=/home/lxp/data/emotic/checkpoints
# CUDA_VISIBLE_DEVICES=0 python vis.py \
# --dataset_file=emotic \
# --num_queries=4 \
# --binary_flag=1 \
# --model=deformable_transformer_dsct \
# --detr=deformable_detr_dsct \
# --data_path=$YOUR_DATA_PATH/images \
# --json_path=./datasets/annotations/emotic_test_bi.json \
# --pretrained_weights=$YOUR_MODEL_PATH/checkpoint.pth;
##### caer-s training scripts #####
YOUR_DATA_PATH=/home/lxp/data/CAER_S
CUDA_VISIBLE_DEVICES=0 python -m torch.distributed.launch \
--nproc_per_node=1 \
--master_port=29507 \
--use_env main.py \
--dataset_file=caer \
--binary_flag=0 \
--detr=deformable_detr_dsct \
--model=deformable_transformer_dsct \
--batch_size=4 \
--cls_loss_coef=5 \
--data_path=$YOUR_DATA_PATH \
--output_dir=$YOUR_DATA_PATH/checkpoints \
--epochs=50 \
--lr_drop=40 \
--num_queries=9 \
--pretrained_weights=./r50_deformable_detr-checkpoint.pth \
# ##### caer-s testing scripts #####
# YOUR_DATA_PATH=/home/lxp/data/CAER_S
# YOUR_MODEL_PATH=/home/lxp/data/CAER_S/checkpoints
# CUDA_VISIBLE_DEVICES=0 python test.py \
# --dataset_file=caer \
# --detr=deformable_detr_dsct \
# --model=deformable_transformer_dsct \
# --num_queries=9 \
# --binary_flag=0 \
# --data_path=$YOUR_DATA_PATH/test \
# --json_path=./datasets/annotations/caer_test.json \
# --pretrained_weights=$YOUR_MODEL_PATH/checkpoint.pth;
# ##### caer-s visualization scripts #####
# YOUR_DATA_PATH=/home/lxp/data/CAER_S
# YOUR_MODEL_PATH=/home/lxp/data/CAER_S/checkpoints
# CUDA_VISIBLE_DEVICES=0 python vis.py \
# --dataset_file=caer \
# --binary_flag=0 \
# --num_queries=9 \
# --model=deformable_transformer_dsct \
# --detr=deformable_detr_dsct \
# --data_path=$YOUR_DATA_PATH/test \
# --json_path=./datasets/annotations/caer_test.json \
# --pretrained_weights=$YOUR_MODEL_PATH/checkpoint.pth;