From 1eb0c252f5f5df5330c94b4814d7159249dd3057 Mon Sep 17 00:00:00 2001 From: jm12138 <2286040843@qq.com> Date: Wed, 31 Mar 2021 22:13:35 +0800 Subject: [PATCH] update 1.0.3 --- README.md | 2 +- ppim/models/rednet.py | 8 ++++---- setup.py | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/README.md b/README.md index e03c425..042d08b 100644 --- a/README.md +++ b/README.md @@ -10,7 +10,7 @@ A PaddlePaddle version image model zoo. * Install by pip: ```shell - $ pip install ppim==1.0.1 -i https://pypi.python.org/pypi + $ pip install ppim==1.0.3 -i https://pypi.python.org/pypi ``` * Install by wheel package:[【Releases Packages】](https://github.com/AgentMaker/Paddle-Image-Models/releases) diff --git a/ppim/models/rednet.py b/ppim/models/rednet.py index a92a10f..6080977 100644 --- a/ppim/models/rednet.py +++ b/ppim/models/rednet.py @@ -28,12 +28,12 @@ } -class involution(nn.Layer): +class Involution(nn.Layer): def __init__(self, channels, kernel_size, stride): - super(involution, self).__init__() + super(Involution, self).__init__() self.kernel_size = kernel_size self.stride = stride self.channels = channels @@ -88,7 +88,7 @@ def __init__(self, super(BottleneckBlock, self).__init__(inplanes, planes, stride, downsample, groups, base_width, dilation, norm_layer) width = int(planes * (base_width / 64.)) * groups - self.conv2 = involution(width, 7, stride) + self.conv2 = Involution(width, 7, stride) class RedNet(resnet.ResNet): @@ -122,7 +122,7 @@ def __init__(self, block, depth, num_classes=1000, with_pool=True): ('bn', nn.BatchNorm2D(self.inplanes // 2)), ('activate', nn.ReLU()) ), - involution(self.inplanes // 2, 3, 1), + Involution(self.inplanes // 2, 3, 1), nn.BatchNorm2D(self.inplanes // 2), nn.ReLU(), nn.Sequential( diff --git a/setup.py b/setup.py index 8f2e6bd..90c0992 100644 --- a/setup.py +++ b/setup.py @@ -1,7 +1,7 @@ from setuptools import setup setup( name='ppim', - version='1.0.1', + version='1.0.3', author='jm12138', author_email='2286040843@qq.com', packages=['ppim', 'ppim.models', 'ppim.units'],