search

leeyeehoo / CSRNet-pytorch

CSRNet: Dilated Convolutional Neural Networks for Understanding the Highly Congested Scenes
642 stars 259 forks source link

Models #5

Open Tikam02 opened 5 years ago

Tikam02 commented 5 years ago

@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].

leeyeehoo commented 5 years ago

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.

Tikam02 commented 5 years ago

@leeyeehoo Thanks got it.