Big re-org, working towards making pip/module as 'timm'

Author: rwightman

inference.py

@@ -11,9 +11,9 @@
 import numpy as np
 import torch
 
-from models import create_model, apply_test_time_pool
-from data import Dataset, create_loader, resolve_data_config
-from utils import AverageMeter
+from timm.models import create_model, apply_test_time_pool
+from timm.data import Dataset, create_loader, resolve_data_config
+from timm.utils import AverageMeter
 
 torch.backends.cudnn.benchmark = True
 
@@ -55,6 +55,9 @@
 def main():
     args = parser.parse_args()
 
+    # might as well try to do something useful...
+    args.pretrained = args.pretrained or not args.checkpoint
+
     # create model
     model = create_model(
         args.model,

optim/adam_tf.py

@@ -0,0 +1,2 @@
+torch~=1.0
+torchvision

requirements.txt

@@ -0,0 +1,55 @@
+""" Setup
+"""
+from setuptools import setup, find_packages
+from codecs import open
+from os import path
+
+here = path.abspath(path.dirname(__file__))
+
+# Get the long description from the README file
+with open(path.join(here, 'README.md'), encoding='utf-8') as f:
+    long_description = f.read()
+
+exec(open('timm/version.py').read())
+setup(
+    name='timm',
+    version=__version__,
+    description='(Unofficial) PyTorch Image Models',
+    long_description=long_description,
+    url='https://github.com/rwightman/pytorch-image-models',
+    author='Ross Wightman',
+    author_email='hello@rwightman.com',
+    classifiers=[  # Optional
+        # How mature is this project? Common values are
+        #   3 - Alpha
+        #   4 - Beta
+        #   5 - Production/Stable
+        'Development Status :: 3 - Alpha',
+        'Intended Audience :: Developers',
+        'Topic :: Software Development :: Build Tools',
+        'License :: OSI Approved :: Apache License',
+        'Programming Language :: Python :: 3.6',
+    ],
+
+    # Note that this is a string of words separated by whitespace, not a list.
+    keywords='pytorch pretrained models efficientnet mobilenetv3 mnasnet',
+
+    # You can just specify package directories manually here if your project is
+    # simple. Or you can use find_packages().
+    #
+    # Alternatively, if you just want to distribute a single Python file, use
+    # the `py_modules` argument instead as follows, which will expect a file
+    # called `my_module.py` to exist:
+    #
+    #   py_modules=["my_module"],
+    #
+    packages=find_packages(exclude=['convert']),
+
+    # This field lists other packages that your project depends on to run.
+    # Any package you put here will be installed by pip when your project is
+    # installed, so they must be valid existing projects.
+    #
+    # For an analysis of "install_requires" vs pip's requirements files see:
+    # https://packaging.python.org/en/latest/requirements.html
+    install_requires=['torch', 'torchvision'],
+)

setup.py

@@ -0,0 +1,2 @@
+from .version import __version__
+from .models import create_model

timm/__init__.py

@@ -1,4 +1,4 @@
-from data.constants import *
+from .constants import *
 
 
 def resolve_data_config(model, args, default_cfg={}, verbose=True):