-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.sh
executable file
·49 lines (34 loc) · 1.42 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
# xyz
#export CUDA_LAUNCH_BLOCKING=1
#export CUDA_VISIBLE_DEVICES=1
CONFIG=configs/strpoints/bev_strpoints_r50_fpn_1x.py
#CONFIG=configs/strpoints/bev_strpoints_r50_fpn_1x_room.py
#CONFIG=configs/strpoints/S3dProj_BevResNet_strpoints_r50_fpn_1x.py
wkdir=bTPV_r50_fpn_XYXYSin2__beike2d_wado_bs6_lr10_LsW510R2P1N1_Rfiou741_Fpn44_Pbs1_Bp32
#wkdir=test
#CONFIG=./work_dirs/${wkdir}/_S3dProj_BevResNet_strpoints_r50_fpn_1x.py
#CONFIG=./work_dirs/${wkdir}/_bev_strpoints_r50_fpn_1x.py
#CP=./work_dirs/${wkdir}/best.pth
#CP=./checkpoints/beike/jun1_wado_bev.pth
#CP=./checkpoints/beike/May4_wd_Bev.pth
#CP=./checkpoints/sfd/15May_Pcl_abcdi_train_6as.pth
LR=0.01
ROTATE=1
BASE_PLANE=32
BS=7
DATA_TYPES=cnx
FILTER_EDGES=0
REL=1
#CLS=abcdif
CLS=ad
#CLS=r
# single gpu
ipython tools/train.py -- ${CONFIG} --rotate $ROTATE --lr $LR --base_plane $BASE_PLANE --bs $BS --data_types $DATA_TYPES --filter_edges $FILTER_EDGES --classes $CLS --relation $REL
#--resume $CP
# Multi gpu
#./tools/dist_train.sh ${CONFIG} 2 --rotate $ROTATE --lr $LR --base_plane $BASE_PLANE --data_types $DATA_TYPES --filter_edges $FILTER_EDGES --classes $CLS --relation $REL
#--resume $CP
ROTATE=0
STYLE='--out ./work_dirs/'${wkdir}'/detection.pickle --eval bbox'
#STYLE=--show
#ipython tools/test.py -- ${CONFIG} $CP --rotate $ROTATE $STYLE --base_plane $BASE_PLANE --data_types $DATA_TYPES --filter_edges $FILTER_EDGES --classes $CLS --relation $REL