yahoo / open_nsfw

Not Suitable for Work (NSFW) classification using deep neural network Caffe models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Why resize to (256, 256) ?

mohanson opened this issue · comments

In deploy.prototxt, the input_param is { shape: { dim: 1 dim: 3 dim: 224 dim: 224 } } but in classify_nsfw.py, images are resized to (256, 256).

@mohanson
In training phase, the image size is 256_256, then random crop size 224_224 to do data augmentation
While in test size, center cropping is uesd to test the image

In this situation, image are resized to 256_256, but caffe only process the center part 224_224

Very thanks and sorry for I didn't take a look in closed issues.