python3 predict.py --model [pretrained model] --dir_demo [demo images directory] --save_name [directory to save] --pre_train [weightfile]
- n_threads: number of threads for data loading
- cpu: use cpu only
- dir_demo: demo image directory
- model: model name
- pre_train: pretrained model directory
- save_name: directory to save