|
a |
|
b/my_tools/amagi_inf.sh |
|
|
1 |
frame_interval="3" |
|
|
2 |
|
|
|
3 |
SPLIT=1 |
|
|
4 |
task_type="active_bleeding" |
|
|
5 |
ckpt="split12_amagi_last.pth" |
|
|
6 |
config_file="configs/recognition/amagi/split1_SF152_ts3_clip8_222.py" |
|
|
7 |
data_list="[txt file that composed of lines of frame folder names]" #split_1_validation.txt" |
|
|
8 |
output_prefix="results/" |
|
|
9 |
|
|
|
10 |
data_path="[base path of dataset]" |
|
|
11 |
num_class="2" |
|
|
12 |
kfold=${SPLIT} |
|
|
13 |
batch_size="64" |
|
|
14 |
|
|
|
15 |
CUDA_VISIBLE_DEVICES=4 python -m torch.distributed.launch --nproc_per_node=1 inference/total_video_inference.py --data_path ${data_path} --batch_size ${batch_size} --kfold ${kfold} --num_class ${num_class} --task_type ${task_type} --frame_interval ${frame_interval} --ckpt ${ckpt} --config_file ${config_file} --data_list ${data_list} --output_prefix ${output_prefix} |