trt_pose
trt_pose copied to clipboard
I want to train mobilenet_v2 to use trt_pose
Hi, To accelerate your pose estimation, I tried training mobilenet_v2. So first, I made I copy densenet.py and changed word 'densenet' to 'mobilenet_v2' including what's in vi editor. and second, I wanted to use train.py (trt_pose/trt_pose/train.py). But there was an error message like that. How should I write to implement train.py ???????
(I typed): python3 train.py --model=mobilenet_v2 usage: train.py [-h] config train.py: error: the following arguments are required: config
you need the json file(config) it's in 'trt_pose/tasks/human_pose/experiments/'