search

yahoo / open_nsfw

Not Suitable for Work (NSFW) classification using deep neural network Caffe models.
BSD 2-Clause "Simplified" License
5.87k stars 1.05k forks source link

Why resize to (256, 256) ? #25

Closed mohanson closed 8 years ago

mohanson commented 8 years ago

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).

chibai commented 8 years ago

@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

mohanson commented 8 years ago

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