RQuispeC / pytorch-ACSCP

Unofficial implementation of "Crowd Counting via Adversarial Cross-Scale Consistency Pursuit" with pytorch - CVPR 2018
MIT License
18 stars 4 forks source link

Is the size of input images for training or testing fixed? #4

Closed wangyuanyuan0 closed 5 years ago

wangyuanyuan0 commented 5 years ago

To maintain the original characteristics of the dataset, can we use the original size image as the model input?

RQuispeC commented 5 years ago

Short answer: Yes, we can use random size images for training and testing

Long answer: The model has been designed to use 256x256 images, however, the code automatically will handle random size images for training and testing.

For training, the data augmentation receives random size images and will augment creating 256x256 patches.

For testing, the code will automatically separate the image in 256x256 patches, applying padding when necessary, then feed them to the net. Finally, the 256x256 output patches are merged to create an output with the original input size.