@@ -26,9 +26,18 @@ class BaseResNet(PretrainedModelMixin):
classifier_layers=["fc6"],
- prepare_func=prepare,
+ prepare_func=self.prepare,
)
+ def prepare(self, x, size=None, *, swap_channels=True, keep_ratio=True):
+ x = prepare(x, size)
+
+ # if not desired, we need to undo it
+ if not swap_channels:
+ x = x[:, :, ::-1]
+ return x
@property
def functions(self):
return super().functions
@@ -21,7 +21,7 @@ install_requires = [
setup(
name=pkg_name,
version=__version__,
- python_requires=">3.6",
+ python_requires=">3.7",
description='Wrapper for various computer vision models (mostly provided by chainer, chainercv, and chainercv2)',
log_description=open(str(cwd / "README.md")).read(),
author='Dimitri Korsch',