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
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
Show 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