forked from Zhuanglong2/T-RODNet
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathslurm_test_ata5
82 lines (59 loc) · 2.62 KB
/
slurm_test_ata5
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
#!/bin/bash
#SBATCH --job-name=yayayaya # Job name
#SBATCH --output=test.%A.txt # Standard output and error log
#SBATCH --nodes=1 # Run all processes on a single node
#SBATCH --ntasks=1 # Run on a single CPU
#SBATCH --mem=40G # Total RAM to be used
#SBATCH --cpus-per-task=64 # Number of CPU cores
#SBATCH --gres=gpu:1 # Number of GPUs (per node)
#SBATCH -p gpu # Use the gpu partition
#SBATCH --time=12:00:00 # Specify the time needed for your experiment
#SBATCH --qos=gpu-8 # To enable the use of up to 8 GPUs
hostname
if [ -d "/tmp/yahia.dalbah/Pickletest" ]
then
echo "Directory exists"
else
mkdir /tmp/yahia.dalbah
cd /tmp/yahia.dalbah
mkdir /tmp/yahia.dalbah/data_root_test
cp /l/users/yahia.dalbah/DATA_ROOT_TMP/* /tmp/yahia.dalbah/data_root_test
# download all zip files and unzip
cd data_root_test
unzip TRAIN_RAD_H.zip
unzip TRAIN_CAM_0.zip
unzip TEST_RAD_H.zip
unzip TRAIN_RAD_H_ANNO.zip
unzip CAM_CALIB.zip
# make folders for data and annotations
mkdir sequences
mkdir annotations
# rename unzipped folders
mv TRAIN_RAD_H sequences/train
mv TRAIN_CAM_0 train
mv TEST_RAD_H sequences/test
mv TRAIN_RAD_H_ANNO annotations/train
# merge folders and remove redundant
rsync -av train/ sequences/train/
rm -r train
python /home/yahia.dalbah/T-RODNet/tools/prepare_dataset/prepare_data.py \
--config /home/yahia.dalbah/T-RODNet/configs/config_rodnet_hg1v2_win16_mnet.py \
--data_root /tmp/yahia.dalbah/data_root_test/ --split train,test --out_data_dir /tmp/yahia.dalbah/Pickletest ;
fi
rm -rf /l/users/yahia.dalbah/results/test5
mkdir /l/users/yahia.dalbah/results/test55
#rm -rf /l/users/yahia.dalbah/results/output_final
mkdir /l/users/yahia.dalbah/results/output_final
mkdir /l/users/yahia.dalbah/zipfiles/
cd /home/yahia.dalbah/T-RODNet/
python /home/yahia.dalbah/T-RODNet/tools/test.py \
--config /home/yahia.dalbah/T-RODNet/configs/config_T_Rodnet_win16.py \
--data_dir /tmp/yahia.dalbah/Pickletest \
--checkpoint /l/users/yahia.dalbah/store/T_RODNet-20230130-230040/epoch_54_final.pkl \
--res_dir /l/users/yahia.dalbah/results/test55/ ;
python /home/yahia.dalbah/RODNet_testing/tools/format_transform/convert_rodnet_to_rod2021-slurm1.py \
--result_dir /l/users/yahia.dalbah/results/test55/T_RODNet-20230130-230040/ \
--final_dir /l/users/yahia.dalbah/results/output_final/test55 ;
cd /l/users/yahia.dalbah/results/output_final/test55/
zip /l/users/yahia.dalbah/zipfiles/5epoch_60_final.zip *.txt
EOL