CSRNet-pytorch icon indicating copy to clipboard operation
CSRNet-pytorch copied to clipboard

Published 20 hours ago • verified publisher icon leeyeehoo

Models

Open Tikam02 opened this issue 7 years ago • 2 comments

@leeyeehoo how to run train and test, problem is what is "task id to use".

~/ML/Crowd-count/final/CSRNet$ python2 train.py TRAIN /home/tikam/ML/Crowd-count/final/CSRNet/train_A.json GPU 0 TASK 0 usage: train.py [-h] [--pre PRETRAINED] TRAIN TEST GPU TASK train.py: error: unrecognized arguments: TASK 1

should default GPU 0 and TASK 0 ---> is it okay

What to use in TASK [ID].

Tikam02 avatar Jul 18 '18 09:07 Tikam02

test path is required... it's the json file which contains the validation images paths. The Task tag is just a label for your saved model. Thus you can just use '0' or '1' or something that can help you distinguish different models you've trained.

Thanks.

leeyeehoo avatar Jul 21 '18 11:07 leeyeehoo

@leeyeehoo Thanks got it.

Tikam02 avatar Jul 21 '18 12:07 Tikam02