-
Notifications
You must be signed in to change notification settings - Fork 8
/
inference.txt
28 lines (28 loc) · 1.64 KB
/
inference.txt
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
--batch_size 4
--checkpoints_dir checkpoint
--ckpt_path checkpoint/kpsfr_finetuned.pth
--custom_worldcup_root dataset/WorldCup_2014_2018
--sfp_finetuned True
--gpu_ids 0
--isTrain False
--loss_mode all
--model_archi KC
--name example/inference
--nms_thres 0.995
--noise_rotate 0.0084
--noise_trans 5.0
--num_objects 4
--pr_thres 5.0
--public_worldcup_root dataset/soccer_worldcup_2014/soccer_data
--resume False
--step_size 200
--template_path assets
--testset test
--train_epochs 1500
--train_lr 0.0001
--train_stage 1
--trainset train
--use_non_local 1
--weight_decay 0.0
--target_video right/2018_Match_Highlights5_clip_00016-1 left/2014_Match_Highlights1_clip_00007-1
--target_image 128 113