Commit efaf9d7b authored by Yuxin Wu's avatar Yuxin Wu

clean-ups

parent 266ce353
...@@ -20,7 +20,7 @@ import inspect ...@@ -20,7 +20,7 @@ import inspect
# add these directories to sys.path here. If the directory is relative to the # add these directories to sys.path here. If the directory is relative to the
# documentation root, use os.path.abspath to make it absolute, like shown here. # documentation root, use os.path.abspath to make it absolute, like shown here.
sys.path.insert(0, os.path.abspath('../')) sys.path.insert(0, os.path.abspath('../'))
os.environ['TENSORPACK_DOC_BUILDING'] = '1' os.environ['DOC_BUILDING'] = '1'
ON_RTD = (os.environ.get('READTHEDOCS') == 'True') ON_RTD = (os.environ.get('READTHEDOCS') == 'True')
...@@ -373,6 +373,7 @@ def autodoc_skip_member(app, what, name, obj, skip, options): ...@@ -373,6 +373,7 @@ def autodoc_skip_member(app, what, name, obj, skip, options):
'set_tower_func', 'set_tower_func',
'TryResumeTraining', 'TryResumeTraining',
'LeakyReLU', 'LeakyReLU',
'PrefetchOnGPUs',
'guided_relu', 'saliency_map', 'get_scalar_var', 'guided_relu', 'saliency_map', 'get_scalar_var',
'prediction_incorrect', 'huber_loss', 'prediction_incorrect', 'huber_loss',
......
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import os as _os import os as _os
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: base.py # File: base.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from abc import ABCMeta from abc import ABCMeta
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: concurrency.py # File: concurrency.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import multiprocessing as mp import multiprocessing as mp
from .base import Callback from .base import Callback
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: graph.py # File: graph.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
""" Graph related callbacks""" """ Graph related callbacks"""
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: group.py # File: group.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from contextlib import contextmanager from contextlib import contextmanager
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: hooks.py # File: hooks.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
""" Compatible layers between tf.train.SessionRunHook and Callback""" """ Compatible layers between tf.train.SessionRunHook and Callback"""
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: inference.py # File: inference.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import numpy as np import numpy as np
from abc import ABCMeta from abc import ABCMeta
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: inference_runner.py # File: inference_runner.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import sys import sys
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: monitor.py # File: monitor.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import os import os
import numpy as np import numpy as np
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: param.py # File: param.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from abc import abstractmethod, ABCMeta from abc import abstractmethod, ABCMeta
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: prof.py # File: prof.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import os import os
import numpy as np import numpy as np
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: saver.py # File: saver.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from datetime import datetime from datetime import datetime
......
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: stats.py # File: stats.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import os import os
import numpy as np import numpy as np
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: summary.py # File: summary.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: trigger.py # File: trigger.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from .base import ProxyCallback, Callback from .base import ProxyCallback, Callback
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: base.py # File: base.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import threading import threading
from abc import abstractmethod, ABCMeta from abc import abstractmethod, ABCMeta
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: common.py # File: common.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from __future__ import division from __future__ import division
import numpy as np import numpy as np
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: bsds500.py # File: bsds500.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import os import os
import glob import glob
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: cifar.py # File: cifar.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
# Yukun Chen <cykustc@gmail.com> # Yukun Chen <cykustc@gmail.com>
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: ilsvrc.py # File: ilsvrc.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import os import os
import tarfile import tarfile
import numpy as np import numpy as np
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: mnist.py # File: mnist.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import os import os
import gzip import gzip
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: svhn.py # File: svhn.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import os import os
import numpy as np import numpy as np
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: dftools.py # File: dftools.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import os import os
import multiprocessing as mp import multiprocessing as mp
......
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: format.py # File: format.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import numpy as np import numpy as np
import six import six
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: image.py # File: image.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import numpy as np import numpy as np
import copy as copy_mod import copy as copy_mod
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import os import os
from pkgutil import iter_modules from pkgutil import iter_modules
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: _test.py # File: _test.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import sys import sys
import cv2 import cv2
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: base.py # File: base.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import inspect import inspect
import pprint import pprint
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: crop.py # File: crop.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from ...utils.argtools import shape2d from ...utils.argtools import shape2d
from .transform import TransformAugmentorBase, CropTransform from .transform import TransformAugmentorBase, CropTransform
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: deform.py # File: deform.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from .base import ImageAugmentor from .base import ImageAugmentor
from ...utils import logger from ...utils import logger
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: geometry.py # File: geometry.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import math import math
import cv2 import cv2
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: imgproc.py # File: imgproc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from .base import ImageAugmentor from .base import ImageAugmentor
from ...utils import logger from ...utils import logger
...@@ -221,16 +221,13 @@ class Saturation(ImageAugmentor): ...@@ -221,16 +221,13 @@ class Saturation(ImageAugmentor):
<https://github.com/facebook/fb.resnet.torch/blob/master/datasets/transforms.lua#L218>`__. <https://github.com/facebook/fb.resnet.torch/blob/master/datasets/transforms.lua#L218>`__.
""" """
def __init__(self, alpha=0.4, rgb=None): def __init__(self, alpha=0.4, rgb=True):
""" """
Args: Args:
alpha(float): maximum saturation change. alpha(float): maximum saturation change.
rgb (bool): whether input is RGB or BGR. rgb (bool): whether input is RGB or BGR.
""" """
super(Saturation, self).__init__() super(Saturation, self).__init__()
if rgb is None:
logger.warn("Saturation() now assumes rgb=False, but will by default use rgb=True in the future!")
rgb = False
rgb = bool(rgb) rgb = bool(rgb)
assert alpha < 1 assert alpha < 1
self._init(locals()) self._init(locals())
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: meta.py # File: meta.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from .base import ImageAugmentor from .base import ImageAugmentor
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: misc.py # File: misc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import numpy as np import numpy as np
import cv2 import cv2
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: noise.py # File: noise.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from .base import ImageAugmentor from .base import ImageAugmentor
import numpy as np import numpy as np
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: paste.py # File: paste.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from .base import ImageAugmentor from .base import ImageAugmentor
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: prefetch.py # File: prefetch.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from __future__ import print_function from __future__ import print_function
import weakref import weakref
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: raw.py # File: raw.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import numpy as np import numpy as np
import copy import copy
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: remote.py # File: remote.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import time import time
import tqdm import tqdm
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: model_desc.py # File: model_desc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
from collections import namedtuple from collections import namedtuple
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: utils.py # File: utils.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from contextlib import contextmanager from contextlib import contextmanager
import operator import operator
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: input_source.py # File: input_source.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
try: try:
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: _test.py # File: _test.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import logging import logging
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: batch_norm.py # File: batch_norm.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from tensorflow.contrib.framework import add_model_variable from tensorflow.contrib.framework import add_model_variable
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: conv2d.py # File: conv2d.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from .common import layer_register, VariableHolder, rename_get_variable from .common import layer_register, VariableHolder, rename_get_variable
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: fc.py # File: fc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: image_sample.py # File: image_sample.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: layer_norm.py # File: layer_norm.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from .common import layer_register from .common import layer_register
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: linearwrap.py # File: linearwrap.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import six import six
from types import ModuleType from types import ModuleType
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: nonlin.py # File: nonlin.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: pool.py # File: pool.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
import numpy as np import numpy as np
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: registry.py # File: registry.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
import inspect import inspect
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: regularize.py # File: regularize.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
import re import re
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: shapes.py # File: shapes.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from .common import layer_register from .common import layer_register
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- File: softmax.py # -*- coding: utf-8 -*- File: softmax.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from .common import layer_register from .common import layer_register
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: base.py # File: base.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from abc import abstractmethod, ABCMeta from abc import abstractmethod, ABCMeta
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: concurrency.py # File: concurrency.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import numpy as np import numpy as np
import multiprocessing import multiprocessing
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: config.py # File: config.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
import six import six
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: dataset.py # File: dataset.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from six.moves import range, zip from six.moves import range, zip
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: multigpu.py # File: multigpu.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from ..utils import logger from ..utils import logger
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: argscope.py # File: argscope.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from contextlib import contextmanager from contextlib import contextmanager
from collections import defaultdict from collections import defaultdict
import inspect import inspect
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: collection.py # File: collection.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from copy import copy from copy import copy
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: common.py # File: common.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from six.moves import map from six.moves import map
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: gradproc.py # File: gradproc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from abc import ABCMeta, abstractmethod from abc import ABCMeta, abstractmethod
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: optimizer.py # File: optimizer.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from contextlib import contextmanager from contextlib import contextmanager
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: scope_utils.py # File: scope_utils.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
import functools import functools
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: sesscreate.py # File: sesscreate.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from .common import get_default_sess_config from .common import get_default_sess_config
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: sessinit.py # File: sessinit.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import os import os
import numpy as np import numpy as np
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: summary.py # File: summary.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import six import six
import tensorflow as tf import tensorflow as tf
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: symbolic_functions.py # File: symbolic_functions.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
from contextlib import contextmanager from contextlib import contextmanager
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: tower.py # File: tower.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
from six.moves import zip from six.moves import zip
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: varmanip.py # File: varmanip.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import six import six
import os import os
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: base.py # File: base.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import time import time
import weakref import weakref
......
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: config.py # File: config.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
__all__ = ['TrainConfig'] __all__ = ['TrainConfig']
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: multigpu.py # File: multigpu.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import tensorflow as tf import tensorflow as tf
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: simple.py # File: simple.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from .base import Trainer from .base import Trainer
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: __init__.py # File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from pkgutil import iter_modules from pkgutil import iter_modules
import os import os
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: concurrency.py # File: concurrency.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
# Credit belongs to Xinyu Zhou # Credit belongs to Xinyu Zhou
import threading import threading
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: debug.py # File: debug.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import sys import sys
......
...@@ -58,7 +58,7 @@ def building_rtfd(): ...@@ -58,7 +58,7 @@ def building_rtfd():
bool: if tensorpack is being imported to generate docs now. bool: if tensorpack is being imported to generate docs now.
""" """
return os.environ.get('READTHEDOCS') == 'True' \ return os.environ.get('READTHEDOCS') == 'True' \
or os.environ.get('TENSORPACK_DOC_BUILDING') or os.environ.get('DOC_BUILDING')
def log_deprecated(name="", text="", eos=""): def log_deprecated(name="", text="", eos=""):
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: fs.py # File: fs.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import os import os
from six.moves import urllib from six.moves import urllib
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: globvars.py # File: globvars.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import six import six
import argparse import argparse
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: gpu.py # File: gpu.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import os import os
from .utils import change_env from .utils import change_env
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: loadcaffe.py # File: loadcaffe.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import sys import sys
import numpy as np import numpy as np
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: logger.py # File: logger.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import logging import logging
import os import os
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: naming.py # File: naming.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import tensorflow as tf import tensorflow as tf
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: rect.py # File: rect.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import numpy as np import numpy as np
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: serialize.py # File: serialize.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import msgpack import msgpack
import msgpack_numpy import msgpack_numpy
...@@ -16,7 +16,7 @@ except ImportError: ...@@ -16,7 +16,7 @@ except ImportError:
__all__ = ['loads', 'dumps'] __all__ = ['loads', 'dumps']
def dumps(obj): def dumps_msgpack(obj):
""" """
Serialize an object. Serialize an object.
Returns: Returns:
...@@ -25,7 +25,7 @@ def dumps(obj): ...@@ -25,7 +25,7 @@ def dumps(obj):
return msgpack.dumps(obj, use_bin_type=True) return msgpack.dumps(obj, use_bin_type=True)
def loads(buf): def loads_msgpack(buf):
""" """
Args: Args:
buf: the output of `dumps`. buf: the output of `dumps`.
...@@ -49,3 +49,7 @@ def loads_pyarrow(buf): ...@@ -49,3 +49,7 @@ def loads_pyarrow(buf):
buf: the output of `dumps`. buf: the output of `dumps`.
""" """
return pa.deserialize(buf) return pa.deserialize(buf)
loads = loads_msgpack
dumps = dumps_msgpack
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: stats.py # File: stats.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import numpy as np import numpy as np
__all__ = ['StatCounter', 'BinaryStatistics', 'RatioCounter', 'Accuracy', __all__ = ['StatCounter', 'BinaryStatistics', 'RatioCounter', 'Accuracy',
......
#!/usr/bin/env python #!/usr/bin/env python
# -*- coding: utf-8 -*- # -*- coding: utf-8 -*-
# File: timer.py # File: timer.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from contextlib import contextmanager from contextlib import contextmanager
......
# -*- coding: UTF-8 -*- # -*- coding: UTF-8 -*-
# File: utils.py # File: utils.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import os import os
import sys import sys
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment