zfnet-512

ZFNet-512

Model	Download	Download (with sample test data)	ONNX version	Opset version	Top-1 accuracy (%)	Top-5 accuracy (%)
ZFNet-512	341 MB	320 MB	1.1	3
ZFNet-512	341 MB	320 MB	1.1.2	6
ZFNet-512	341 MB	320 MB	1.2	7
ZFNet-512	341 MB	318 MB	1.3	8
ZFNet-512	341 MB	318 MB	1.4	9
ZFNet-512	333 MB	309 MB	1.9	12	55.97	79.41
ZFNet-512-int8	83 MB	48 MB	1.9	12	55.84	79.33
ZFNet-512-qdq	84 MB	56 MB	1.9	12	55.83	79.42

Compared with the fp32 ZFNet-512, int8 ZFNet-512's Top-1 accuracy drop ratio is 0.23%, Top-5 accuracy drop ratio is 0.10% and performance improvement is 1.78x.

Note

Different preprocess methods will lead to different accuracies, the accuracy in table depends on this specific preprocess method.

The performance depends on the test hardware. Performance data here is collected with Intel® Xeon® Platinum 8280 Processor, 1s 4c per instance, CentOS Linux 8.3, data batch size is 1.

Description

ZFNet-512 is a deep convolutional networks for classification. This model's 4th layer has 512 maps instead of 1024 maps mentioned in the paper.

Dataset

ILSVRC2013

Source

Caffe2 ZFNet-512 ==> ONNX ZFNet-512

Model input and output

Input

gpu_0/data_0: float[1, 3, 224, 224]

Output

gpu_0/softmax_1: float[1, 1000]

Pre-processing steps

Post-processing steps

Sample test data

random generated sampe test data:

• test_data_set_0
• test_data_set_1
• test_data_set_2
• test_data_set_3
• test_data_set_4
• test_data_set_5

Results/accuracy on test set

Quantization

ZFNet-512-int8 and ZFNet-512-qdq are obtained by quantizing fp32 ZFNet-512 model. We use Intel® Neural Compressor with onnxruntime backend to perform quantization. View the instructions to understand how to use Intel® Neural Compressor for quantization.

Environment

onnx: 1.9.0 onnxruntime: 1.8.0

Prepare model

wget https://github.com/onnx/models/raw/main/vision/classification/zfnet-512/model/zfnet512-12.onnx

Model quantize

Make sure to specify the appropriate dataset path in the configuration file.

bash run_tuning.sh --input_model=path/to/model \  # model path as *.onnx
                   --config=zfnet512.yaml \
                   --data_path=/path/to/imagenet \
                   --label_path=/path/to/imagenet/label \
                   --output_model=path/to/save

References

• Visualizing and Understanding Convolutional Networks

• Intel® Neural Compressor

Contributors

• mengniwang95 (Intel)
• yuwenzho (Intel)
• airMeng (Intel)
• ftian1 (Intel)
• hshen14 (Intel)

License

MIT