Caffe学习：pycaffe利用caffemodel进行分类=>裁剪图片

最新推荐文章于 2024-07-17 10:10:44 发布

jiarenyf

最新推荐文章于 2024-07-17 10:10:44 发布

阅读量9.7k

点赞数

本文链接：https://blog.csdn.net/u011762313/article/details/48343799

版权

本文介绍了在使用Caffe进行图像分类时，如何利用pycaffe结合预训练的caffemodel对大于crop_size的图片进行裁剪以提高分类准确性。在测试阶段，原始代码只会截取图片中心部分，而通过手动裁剪和过采样（oversample）方法，可以选取图片中央、角落及镜像共10个区域，从而提升模型的识别效果。

摘要由CSDN通过智能技术生成

在caffe中，如果定义了crop_size，那么在train时会对大于crop_size的图片进行随机裁剪，而在test时只是截取中间部分（详见/caffe/src/caffe/data_transformer.cpp）：

   // We only do random crop when we do training.
    if (phase_ == TRAIN) {
      h_off = Rand(datum_height - crop_size + 1);
      w_off = Rand(datum_width - crop_size + 1);
    } else {
      h_off = (datum_height - crop_size) / 2;
      w_off = (datum_width - crop_size) / 2;
    }
  }