pickle - 为 Caffe 的 python 分类.py 提供平均像素值

Question

我想用 Python 包装器测试一个 Caffe 模型：

python classify.py --model_del ./deploy.prototxt --pretrained_model ./mymodel.caffemodel input.png output

有没有一种简单的方法可以为mean_pixelpython 包装器赋值？它似乎只支持一个mean_file论点？

Answer 1

该代码使用args.mean_filevariable 将 numpy 格式的数据读取到 variable mean。最简单的方法是引入一个名为的新解析器参数args.mean_pixel，该参数具有单个平均值，将其存储为一个mean_pixel变量，然后创建一个名为的数组，该数组mean具有与输入数据相同的维度，并将该mean_pixel值复制到大批。其余代码将正常运行。

parser.add_argument(
    "--mean_pixel",
    type=float,
    default=128.0,
    help="Enter the mean pixel value to be subtracted."
)

上面的代码段将尝试采用名为mean_pixel.

替换代码段：

if args.mean_file:
    mean = np.load(args.mean_file)

和：

if args.mean_file:
    mean = np.load(args.mean_file)
elif args.mean_pixel:
    mean_pixel = args.mean_pixel
    mean = np.array([image_dims[0],image_dims[1],channels]) #where channels is the number of channels of the image
    mean.fill(mean_pixel)

mean_pixel如果mean_file没有作为参数传递，这将使代码选择作为参数传递的值。上面的代码将创建一个数组，其尺寸与图像的尺寸相同，并用mean_pixel值填充它。

其余代码无需更改。