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
73ac38c7
Commit
73ac38c7
authored
May 23, 2016
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
misc.
parent
90a74f02
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
17 additions
and
11 deletions
+17
-11
README.md
README.md
+8
-6
examples/load-alexnet.py
examples/load-alexnet.py
+6
-0
examples/load-vgg16.py
examples/load-vgg16.py
+2
-2
tensorpack/dataflow/dataset/atari.py
tensorpack/dataflow/dataset/atari.py
+1
-3
No files found.
README.md
View file @
73ac38c7
# tensorpack
Neural Network Toolbox on TensorFlow
In development. API might change a bit. See
[
examples
](
https://github.com/ppwwyyxx/tensorpack/tree/master/examples
)
to learn.
In development. API might change a bit.
See
[
examples
](
https://github.com/ppwwyyxx/tensorpack/tree/master/examples
)
to learn.
## Features:
+
Scoped abstraction of common models.
+
Use
`Dataflow`
to define data preprocessing in pure Python.
+
Callbacks systems to control training.
+
Training and testing are described together. Just need to follow the conventions to setup stuffs.
+
Write summary easier for tensorboard.
Focused on modularity:
+
Models has Scoped abstraction of common models.
+
Dataflow defines data preprocessing in pure Python.
+
Callbacks systems to control training behavior.
examples/load-alexnet.py
View file @
73ac38c7
...
...
@@ -21,6 +21,12 @@ from tensorpack.callbacks import *
from
tensorpack.dataflow
import
*
from
tensorpack.dataflow.dataset
import
ILSVRCMeta
"""
Usage:
python2 -m tensorpack.utils.loadcaffe PATH/TO/CAFFE/{deploy.prototxt,bvlc_alexnet.caffemodel} alexnet.npy
./load-alexnet.py --load alexnet.npy --input cat.png
"""
BATCH_SIZE
=
10
MIN_AFTER_DEQUEUE
=
500
CAPACITY
=
MIN_AFTER_DEQUEUE
+
3
*
BATCH_SIZE
...
...
examples/load-vgg16.py
View file @
73ac38c7
...
...
@@ -23,8 +23,8 @@ from tensorpack.dataflow.dataset import ILSVRCMeta
"""
Usage:
python2 -m tensorpack.utils.loadcaffe PATH/TO/
models/
VGG/{VGG_ILSVRC_16_layers_deploy.prototxt,VGG_ILSVRC_16_layers.caffemodel} vgg16.npy
./load
_
vgg16.py --load vgg16.npy --input cat.png
python2 -m tensorpack.utils.loadcaffe PATH/TO/VGG/{VGG_ILSVRC_16_layers_deploy.prototxt,VGG_ILSVRC_16_layers.caffemodel} vgg16.npy
./load
-
vgg16.py --load vgg16.npy --input cat.png
"""
class
Model
(
ModelDesc
):
...
...
tensorpack/dataflow/dataset/atari.py
View file @
73ac38c7
...
...
@@ -168,10 +168,8 @@ class AtariPlayer(RLEnvironment):
def
get_stat
(
self
):
try
:
print
self
.
stats
return
{
'avg_score'
:
np
.
mean
(
self
.
stats
[
'score'
]),
'max_score'
:
float
(
np
.
max
(
self
.
stats
[
'score'
]))
}
'max_score'
:
float
(
np
.
max
(
self
.
stats
[
'score'
]))
}
except
ValueError
:
return
{}
...
...
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