Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
S
seminar-breakout
Project overview
Project overview
Details
Activity
Releases
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Analytics
Analytics
CI / CD
Repository
Value Stream
Wiki
Wiki
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Shashank Suhas
seminar-breakout
Commits
efaf9d7b
You need to sign in or sign up before continuing.
Commit
efaf9d7b
authored
Jan 11, 2018
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
clean-ups
parent
266ce353
Changes
102
Expand all
Hide whitespace changes
Inline
Side-by-side
Showing
102 changed files
with
106 additions
and
108 deletions
+106
-108
docs/conf.py
docs/conf.py
+2
-1
tensorpack/__init__.py
tensorpack/__init__.py
+1
-1
tensorpack/callbacks/__init__.py
tensorpack/callbacks/__init__.py
+1
-1
tensorpack/callbacks/base.py
tensorpack/callbacks/base.py
+1
-1
tensorpack/callbacks/concurrency.py
tensorpack/callbacks/concurrency.py
+1
-1
tensorpack/callbacks/graph.py
tensorpack/callbacks/graph.py
+1
-1
tensorpack/callbacks/group.py
tensorpack/callbacks/group.py
+1
-1
tensorpack/callbacks/hooks.py
tensorpack/callbacks/hooks.py
+1
-1
tensorpack/callbacks/inference.py
tensorpack/callbacks/inference.py
+1
-1
tensorpack/callbacks/inference_runner.py
tensorpack/callbacks/inference_runner.py
+1
-1
tensorpack/callbacks/monitor.py
tensorpack/callbacks/monitor.py
+1
-1
tensorpack/callbacks/param.py
tensorpack/callbacks/param.py
+1
-1
tensorpack/callbacks/prof.py
tensorpack/callbacks/prof.py
+1
-1
tensorpack/callbacks/saver.py
tensorpack/callbacks/saver.py
+1
-1
tensorpack/callbacks/stats.py
tensorpack/callbacks/stats.py
+1
-1
tensorpack/callbacks/summary.py
tensorpack/callbacks/summary.py
+1
-1
tensorpack/callbacks/trigger.py
tensorpack/callbacks/trigger.py
+1
-1
tensorpack/dataflow/__init__.py
tensorpack/dataflow/__init__.py
+1
-1
tensorpack/dataflow/base.py
tensorpack/dataflow/base.py
+1
-1
tensorpack/dataflow/common.py
tensorpack/dataflow/common.py
+1
-1
tensorpack/dataflow/dataset/__init__.py
tensorpack/dataflow/dataset/__init__.py
+1
-1
tensorpack/dataflow/dataset/bsds500.py
tensorpack/dataflow/dataset/bsds500.py
+1
-1
tensorpack/dataflow/dataset/cifar.py
tensorpack/dataflow/dataset/cifar.py
+1
-1
tensorpack/dataflow/dataset/ilsvrc.py
tensorpack/dataflow/dataset/ilsvrc.py
+1
-1
tensorpack/dataflow/dataset/mnist.py
tensorpack/dataflow/dataset/mnist.py
+1
-1
tensorpack/dataflow/dataset/svhn.py
tensorpack/dataflow/dataset/svhn.py
+1
-1
tensorpack/dataflow/dftools.py
tensorpack/dataflow/dftools.py
+1
-1
tensorpack/dataflow/format.py
tensorpack/dataflow/format.py
+1
-1
tensorpack/dataflow/image.py
tensorpack/dataflow/image.py
+1
-1
tensorpack/dataflow/imgaug/__init__.py
tensorpack/dataflow/imgaug/__init__.py
+1
-1
tensorpack/dataflow/imgaug/_test.py
tensorpack/dataflow/imgaug/_test.py
+1
-1
tensorpack/dataflow/imgaug/base.py
tensorpack/dataflow/imgaug/base.py
+1
-1
tensorpack/dataflow/imgaug/crop.py
tensorpack/dataflow/imgaug/crop.py
+1
-1
tensorpack/dataflow/imgaug/deform.py
tensorpack/dataflow/imgaug/deform.py
+1
-1
tensorpack/dataflow/imgaug/geometry.py
tensorpack/dataflow/imgaug/geometry.py
+1
-1
tensorpack/dataflow/imgaug/imgproc.py
tensorpack/dataflow/imgaug/imgproc.py
+2
-5
tensorpack/dataflow/imgaug/meta.py
tensorpack/dataflow/imgaug/meta.py
+0
-1
tensorpack/dataflow/imgaug/misc.py
tensorpack/dataflow/imgaug/misc.py
+1
-1
tensorpack/dataflow/imgaug/noise.py
tensorpack/dataflow/imgaug/noise.py
+1
-1
tensorpack/dataflow/imgaug/paste.py
tensorpack/dataflow/imgaug/paste.py
+1
-1
tensorpack/dataflow/prefetch.py
tensorpack/dataflow/prefetch.py
+1
-1
tensorpack/dataflow/raw.py
tensorpack/dataflow/raw.py
+1
-1
tensorpack/dataflow/remote.py
tensorpack/dataflow/remote.py
+1
-1
tensorpack/graph_builder/__init__.py
tensorpack/graph_builder/__init__.py
+1
-1
tensorpack/graph_builder/model_desc.py
tensorpack/graph_builder/model_desc.py
+1
-1
tensorpack/graph_builder/utils.py
tensorpack/graph_builder/utils.py
+1
-1
tensorpack/input_source/__init__.py
tensorpack/input_source/__init__.py
+1
-1
tensorpack/input_source/input_source.py
tensorpack/input_source/input_source.py
+1
-1
tensorpack/models/__init__.py
tensorpack/models/__init__.py
+1
-1
tensorpack/models/_test.py
tensorpack/models/_test.py
+1
-1
tensorpack/models/batch_norm.py
tensorpack/models/batch_norm.py
+1
-1
tensorpack/models/conv2d.py
tensorpack/models/conv2d.py
+1
-1
tensorpack/models/fc.py
tensorpack/models/fc.py
+1
-1
tensorpack/models/image_sample.py
tensorpack/models/image_sample.py
+1
-1
tensorpack/models/layer_norm.py
tensorpack/models/layer_norm.py
+1
-1
tensorpack/models/linearwrap.py
tensorpack/models/linearwrap.py
+1
-1
tensorpack/models/nonlin.py
tensorpack/models/nonlin.py
+1
-1
tensorpack/models/pool.py
tensorpack/models/pool.py
+1
-1
tensorpack/models/registry.py
tensorpack/models/registry.py
+1
-1
tensorpack/models/regularize.py
tensorpack/models/regularize.py
+1
-1
tensorpack/models/shapes.py
tensorpack/models/shapes.py
+1
-1
tensorpack/models/softmax.py
tensorpack/models/softmax.py
+1
-1
tensorpack/predict/__init__.py
tensorpack/predict/__init__.py
+1
-1
tensorpack/predict/base.py
tensorpack/predict/base.py
+1
-1
tensorpack/predict/concurrency.py
tensorpack/predict/concurrency.py
+1
-1
tensorpack/predict/config.py
tensorpack/predict/config.py
+1
-1
tensorpack/predict/dataset.py
tensorpack/predict/dataset.py
+1
-1
tensorpack/predict/multigpu.py
tensorpack/predict/multigpu.py
+1
-1
tensorpack/tfutils/__init__.py
tensorpack/tfutils/__init__.py
+1
-1
tensorpack/tfutils/argscope.py
tensorpack/tfutils/argscope.py
+1
-1
tensorpack/tfutils/collection.py
tensorpack/tfutils/collection.py
+1
-1
tensorpack/tfutils/common.py
tensorpack/tfutils/common.py
+1
-1
tensorpack/tfutils/gradproc.py
tensorpack/tfutils/gradproc.py
+1
-1
tensorpack/tfutils/optimizer.py
tensorpack/tfutils/optimizer.py
+1
-1
tensorpack/tfutils/scope_utils.py
tensorpack/tfutils/scope_utils.py
+1
-1
tensorpack/tfutils/sesscreate.py
tensorpack/tfutils/sesscreate.py
+1
-1
tensorpack/tfutils/sessinit.py
tensorpack/tfutils/sessinit.py
+1
-1
tensorpack/tfutils/summary.py
tensorpack/tfutils/summary.py
+1
-1
tensorpack/tfutils/symbolic_functions.py
tensorpack/tfutils/symbolic_functions.py
+1
-1
tensorpack/tfutils/tower.py
tensorpack/tfutils/tower.py
+1
-1
tensorpack/tfutils/varmanip.py
tensorpack/tfutils/varmanip.py
+1
-1
tensorpack/train/__init__.py
tensorpack/train/__init__.py
+1
-1
tensorpack/trainv1/__init__.py
tensorpack/trainv1/__init__.py
+1
-1
tensorpack/trainv1/base.py
tensorpack/trainv1/base.py
+1
-1
tensorpack/trainv1/config.py
tensorpack/trainv1/config.py
+1
-1
tensorpack/trainv1/multigpu.py
tensorpack/trainv1/multigpu.py
+1
-1
tensorpack/trainv1/simple.py
tensorpack/trainv1/simple.py
+0
-1
tensorpack/utils/__init__.py
tensorpack/utils/__init__.py
+1
-1
tensorpack/utils/concurrency.py
tensorpack/utils/concurrency.py
+1
-1
tensorpack/utils/debug.py
tensorpack/utils/debug.py
+0
-1
tensorpack/utils/develop.py
tensorpack/utils/develop.py
+1
-1
tensorpack/utils/fs.py
tensorpack/utils/fs.py
+1
-1
tensorpack/utils/globvars.py
tensorpack/utils/globvars.py
+1
-1
tensorpack/utils/gpu.py
tensorpack/utils/gpu.py
+1
-1
tensorpack/utils/loadcaffe.py
tensorpack/utils/loadcaffe.py
+1
-1
tensorpack/utils/logger.py
tensorpack/utils/logger.py
+1
-1
tensorpack/utils/naming.py
tensorpack/utils/naming.py
+1
-1
tensorpack/utils/rect.py
tensorpack/utils/rect.py
+1
-1
tensorpack/utils/serialize.py
tensorpack/utils/serialize.py
+7
-3
tensorpack/utils/stats.py
tensorpack/utils/stats.py
+1
-1
tensorpack/utils/timer.py
tensorpack/utils/timer.py
+0
-1
tensorpack/utils/utils.py
tensorpack/utils/utils.py
+1
-1
No files found.
docs/conf.py
View file @
efaf9d7b
...
...
@@ -20,7 +20,7 @@ import inspect
# 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.
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'
)
...
...
@@ -373,6 +373,7 @@ def autodoc_skip_member(app, what, name, obj, skip, options):
'set_tower_func'
,
'TryResumeTraining'
,
'LeakyReLU'
,
'PrefetchOnGPUs'
,
'guided_relu'
,
'saliency_map'
,
'get_scalar_var'
,
'prediction_incorrect'
,
'huber_loss'
,
...
...
tensorpack/__init__.py
View file @
efaf9d7b
# -*- coding: utf-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
os
as
_os
...
...
tensorpack/callbacks/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/callbacks/base.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: base.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
abc
import
ABCMeta
...
...
tensorpack/callbacks/concurrency.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: concurrency.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
multiprocessing
as
mp
from
.base
import
Callback
...
...
tensorpack/callbacks/graph.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: graph.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
""" Graph related callbacks"""
...
...
tensorpack/callbacks/group.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: group.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
contextlib
import
contextmanager
...
...
tensorpack/callbacks/hooks.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: hooks.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
""" Compatible layers between tf.train.SessionRunHook and Callback"""
...
...
tensorpack/callbacks/inference.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: inference.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
numpy
as
np
from
abc
import
ABCMeta
...
...
tensorpack/callbacks/inference_runner.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: inference_runner.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
sys
import
tensorflow
as
tf
...
...
tensorpack/callbacks/monitor.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: monitor.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
os
import
numpy
as
np
...
...
tensorpack/callbacks/param.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: param.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
abc
import
abstractmethod
,
ABCMeta
...
...
tensorpack/callbacks/prof.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: prof.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
os
import
numpy
as
np
...
...
tensorpack/callbacks/saver.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: saver.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
datetime
import
datetime
...
...
tensorpack/callbacks/stats.py
View file @
efaf9d7b
# -*- coding: utf-8 -*-
# File: stats.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
os
import
numpy
as
np
...
...
tensorpack/callbacks/summary.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: summary.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
...
...
tensorpack/callbacks/trigger.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: trigger.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
.base
import
ProxyCallback
,
Callback
...
...
tensorpack/dataflow/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/dataflow/base.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: base.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
threading
from
abc
import
abstractmethod
,
ABCMeta
...
...
tensorpack/dataflow/common.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: common.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
__future__
import
division
import
numpy
as
np
...
...
tensorpack/dataflow/dataset/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/dataflow/dataset/bsds500.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: bsds500.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
os
import
glob
...
...
tensorpack/dataflow/dataset/cifar.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: cifar.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
# Yukun Chen <cykustc@gmail.com>
import
os
...
...
tensorpack/dataflow/dataset/ilsvrc.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: ilsvrc.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
os
import
tarfile
import
numpy
as
np
...
...
tensorpack/dataflow/dataset/mnist.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: mnist.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
os
import
gzip
...
...
tensorpack/dataflow/dataset/svhn.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: svhn.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
os
import
numpy
as
np
...
...
tensorpack/dataflow/dftools.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: dftools.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
os
import
multiprocessing
as
mp
...
...
tensorpack/dataflow/format.py
View file @
efaf9d7b
# -*- coding: utf-8 -*-
# File: format.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
numpy
as
np
import
six
...
...
tensorpack/dataflow/image.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: image.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
numpy
as
np
import
copy
as
copy_mod
...
...
tensorpack/dataflow/imgaug/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
os
from
pkgutil
import
iter_modules
...
...
tensorpack/dataflow/imgaug/_test.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: _test.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
sys
import
cv2
...
...
tensorpack/dataflow/imgaug/base.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: base.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
inspect
import
pprint
...
...
tensorpack/dataflow/imgaug/crop.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: crop.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
...utils.argtools
import
shape2d
from
.transform
import
TransformAugmentorBase
,
CropTransform
...
...
tensorpack/dataflow/imgaug/deform.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: deform.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
.base
import
ImageAugmentor
from
...utils
import
logger
...
...
tensorpack/dataflow/imgaug/geometry.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: geometry.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
math
import
cv2
...
...
tensorpack/dataflow/imgaug/imgproc.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: imgproc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
.base
import
ImageAugmentor
from
...utils
import
logger
...
...
@@ -221,16 +221,13 @@ class Saturation(ImageAugmentor):
<https://github.com/facebook/fb.resnet.torch/blob/master/datasets/transforms.lua#L218>`__.
"""
def
__init__
(
self
,
alpha
=
0.4
,
rgb
=
Non
e
):
def
__init__
(
self
,
alpha
=
0.4
,
rgb
=
Tru
e
):
"""
Args:
alpha(float): maximum saturation change.
rgb (bool): whether input is RGB or BGR.
"""
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
)
assert
alpha
<
1
self
.
_init
(
locals
())
...
...
tensorpack/dataflow/imgaug/meta.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: meta.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
.base
import
ImageAugmentor
...
...
tensorpack/dataflow/imgaug/misc.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: misc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
numpy
as
np
import
cv2
...
...
tensorpack/dataflow/imgaug/noise.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: noise.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
.base
import
ImageAugmentor
import
numpy
as
np
...
...
tensorpack/dataflow/imgaug/paste.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: paste.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
.base
import
ImageAugmentor
...
...
tensorpack/dataflow/prefetch.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: prefetch.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
__future__
import
print_function
import
weakref
...
...
tensorpack/dataflow/raw.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: raw.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
numpy
as
np
import
copy
...
...
tensorpack/dataflow/remote.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: remote.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
time
import
tqdm
...
...
tensorpack/graph_builder/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/graph_builder/model_desc.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: model_desc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
abc
import
ABCMeta
,
abstractmethod
from
collections
import
namedtuple
...
...
tensorpack/graph_builder/utils.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: utils.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
contextlib
import
contextmanager
import
operator
...
...
tensorpack/input_source/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/input_source/input_source.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: input_source.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
try
:
...
...
tensorpack/models/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/models/_test.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: _test.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
logging
import
tensorflow
as
tf
...
...
tensorpack/models/batch_norm.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: batch_norm.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
tensorflow.contrib.framework
import
add_model_variable
...
...
tensorpack/models/conv2d.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: conv2d.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
.common
import
layer_register
,
VariableHolder
,
rename_get_variable
...
...
tensorpack/models/fc.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: fc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
...
...
tensorpack/models/image_sample.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: image_sample.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
...
...
tensorpack/models/layer_norm.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: layer_norm.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
.common
import
layer_register
...
...
tensorpack/models/linearwrap.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: linearwrap.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
six
from
types
import
ModuleType
...
...
tensorpack/models/nonlin.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: nonlin.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
...
...
tensorpack/models/pool.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: pool.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
import
numpy
as
np
...
...
tensorpack/models/registry.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: registry.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
import
inspect
...
...
tensorpack/models/regularize.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: regularize.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
import
re
...
...
tensorpack/models/shapes.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: shapes.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
.common
import
layer_register
...
...
tensorpack/models/softmax.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*- File: softmax.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
.common
import
layer_register
...
...
tensorpack/predict/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/predict/base.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: base.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
abc
import
abstractmethod
,
ABCMeta
import
tensorflow
as
tf
...
...
tensorpack/predict/concurrency.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: concurrency.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
numpy
as
np
import
multiprocessing
...
...
tensorpack/predict/config.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: config.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
import
six
...
...
tensorpack/predict/dataset.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: dataset.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
six.moves
import
range
,
zip
from
abc
import
ABCMeta
,
abstractmethod
...
...
tensorpack/predict/multigpu.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: multigpu.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
..utils
import
logger
...
...
tensorpack/tfutils/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/tfutils/argscope.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: argscope.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
from
contextlib
import
contextmanager
from
collections
import
defaultdict
import
inspect
...
...
tensorpack/tfutils/collection.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: collection.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
copy
import
copy
...
...
tensorpack/tfutils/common.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: common.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
six.moves
import
map
...
...
tensorpack/tfutils/gradproc.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
# File: gradproc.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
abc
import
ABCMeta
,
abstractmethod
...
...
tensorpack/tfutils/optimizer.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: optimizer.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
contextlib
import
contextmanager
...
...
tensorpack/tfutils/scope_utils.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: scope_utils.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
import
functools
...
...
tensorpack/tfutils/sesscreate.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: sesscreate.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
.common
import
get_default_sess_config
...
...
tensorpack/tfutils/sessinit.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: sessinit.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
os
import
numpy
as
np
...
...
tensorpack/tfutils/summary.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: summary.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
six
import
tensorflow
as
tf
...
...
tensorpack/tfutils/symbolic_functions.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: symbolic_functions.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
from
contextlib
import
contextmanager
...
...
tensorpack/tfutils/tower.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: tower.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
from
six.moves
import
zip
...
...
tensorpack/tfutils/varmanip.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: varmanip.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
six
import
os
...
...
tensorpack/train/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/trainv1/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/trainv1/base.py
View file @
efaf9d7b
# -*- coding: utf-8 -*-
# File: base.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
time
import
weakref
...
...
tensorpack/trainv1/config.py
View file @
efaf9d7b
# -*- coding: utf-8 -*-
# File: config.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
__all__
=
[
'TrainConfig'
]
...
...
tensorpack/trainv1/multigpu.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: multigpu.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
tensorflow
as
tf
...
...
tensorpack/trainv1/simple.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: simple.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
.base
import
Trainer
...
...
tensorpack/utils/__init__.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: __init__.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
from
pkgutil
import
iter_modules
import
os
...
...
tensorpack/utils/concurrency.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: concurrency.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
# Credit belongs to Xinyu Zhou
import
threading
...
...
tensorpack/utils/debug.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: debug.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
sys
...
...
tensorpack/utils/develop.py
View file @
efaf9d7b
...
...
@@ -58,7 +58,7 @@ def building_rtfd():
bool: if tensorpack is being imported to generate docs now.
"""
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
=
""
):
...
...
tensorpack/utils/fs.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: fs.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
os
from
six.moves
import
urllib
...
...
tensorpack/utils/globvars.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: globvars.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
six
import
argparse
...
...
tensorpack/utils/gpu.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: gpu.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
os
from
.utils
import
change_env
...
...
tensorpack/utils/loadcaffe.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: loadcaffe.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
sys
import
numpy
as
np
...
...
tensorpack/utils/logger.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: logger.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
logging
import
os
...
...
tensorpack/utils/naming.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: naming.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
tensorflow
as
tf
...
...
tensorpack/utils/rect.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: rect.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
numpy
as
np
...
...
tensorpack/utils/serialize.py
View file @
efaf9d7b
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# File: serialize.py
# Author: Yuxin Wu <ppwwyyxxc@gmail.com>
import
msgpack
import
msgpack_numpy
...
...
@@ -16,7 +16,7 @@ except ImportError:
__all__
=
[
'loads'
,
'dumps'
]
def
dumps
(
obj
):
def
dumps
_msgpack
(
obj
):
"""
Serialize an object.
Returns:
...
...
@@ -25,7 +25,7 @@ def dumps(obj):
return
msgpack
.
dumps
(
obj
,
use_bin_type
=
True
)
def
loads
(
buf
):
def
loads
_msgpack
(
buf
):
"""
Args:
buf: the output of `dumps`.
...
...
@@ -49,3 +49,7 @@ def loads_pyarrow(buf):
buf: the output of `dumps`.
"""
return
pa
.
deserialize
(
buf
)
loads
=
loads_msgpack
dumps
=
dumps_msgpack
tensorpack/utils/stats.py
View file @
efaf9d7b
# -*- coding: UTF-8 -*-
# File: stats.py
# Author: Yuxin Wu <ppwwyyxx@gmail.com>
import
numpy
as
np
__all__
=
[
'StatCounter'
,
'BinaryStatistics'
,
'RatioCounter'
,
'Accuracy'
,
...
...
tensorpack/utils/timer.py
View file @
efaf9d7b
This diff is collapsed.
Click to expand it.
tensorpack/utils/utils.py
View file @
efaf9d7b
This diff is collapsed.
Click to expand it.
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment