leeyeehoo / CSRNet-pytorch

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

Region of Interest at test time #14

Closed IssamLaradji closed 6 years ago

IssamLaradji commented 6 years ago

Great code thank you!

For datasets with regions of interest (ROI) such as UCSD and world Expo, do you use the ROI to mask the input image at test time? I think it makes sense that the ROI is accessible at test time (as it is unique to each scene/camera). Thanks.

leeyeehoo commented 6 years ago

Yes. We first put ROI masks on the training dataset. And test with ROI. Thanks.

IssamLaradji commented 6 years ago

Thanks!