From 2ca86c4d4822b893fc5a131af9c08952dd724ebb Mon Sep 17 00:00:00 2001
From: "xiuquan.yan" <xiuquan.yan@roadefend.com>
Date: Wed, 11 Dec 2019 11:01:45 +0000
Subject: [PATCH] support AdaptiveAvgPool2d

---
 pytorch_to_caffe.py | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/pytorch_to_caffe.py b/pytorch_to_caffe.py
index d58aec1..94cd6a1 100755
--- a/pytorch_to_caffe.py
+++ b/pytorch_to_caffe.py
@@ -191,6 +191,19 @@ def _avg_pool2d(raw,input, kernel_size, stride = None, padding = 0, ceil_mode =
     _pool('ave',raw,input, x, kernel_size, stride, padding,ceil_mode)
     return x
 
+def _adaptive_avg_pool2d(raw, input, output_size):
+    _output_size = _list_with_default(output_size, input.size())
+    x = raw(input, _output_size)
+    if isinstance(_output_size, int):
+        out_dim = _output_size
+    else:
+        out_dim = _output_size[0]
+    tmp = max(input.shape[2], input.shape[3])
+    stride = tmp //out_dim
+    kernel_size = tmp - (out_dim - 1) * stride
+    _pool('ave', raw, input, x, kernel_size, stride, 0, False)
+    return x
+
 def _max(raw,*args):
     x=raw(*args)
     if len(args)==1: