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
8bc909be
Commit
8bc909be
authored
Nov 14, 2017
by
Yuxin Wu
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
small update in docs
parent
09995c03
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
9 deletions
+8
-9
examples/FasterRCNN/train.py
examples/FasterRCNN/train.py
+2
-3
examples/README.md
examples/README.md
+6
-6
No files found.
examples/FasterRCNN/train.py
View file @
8bc909be
...
...
@@ -103,7 +103,7 @@ class Model(ModelDesc):
boxes_on_featuremap
=
proposal_boxes
*
(
1.0
/
config
.
ANCHOR_STRIDE
)
roi_resized
=
roi_align
(
featuremap
,
boxes_on_featuremap
,
14
)
feature_fastrcnn
=
resnet_conv5
(
roi_resized
,
config
.
RESNET_NUM_BLOCK
[
-
1
])
# nxc
feature_fastrcnn
=
resnet_conv5
(
roi_resized
,
config
.
RESNET_NUM_BLOCK
[
-
1
])
# nxc
x7x7
fastrcnn_label_logits
,
fastrcnn_box_logits
=
fastrcnn_head
(
'fastrcnn'
,
feature_fastrcnn
,
config
.
NUM_CLASS
)
if
is_training
:
...
...
@@ -133,8 +133,7 @@ class Model(ModelDesc):
fastrcnn_label_loss
,
fastrcnn_box_loss
,
wd_cost
],
'total_cost'
)
for
k
in
self
.
cost
,
wd_cost
:
add_moving_summary
(
k
)
add_moving_summary
(
self
.
cost
,
wd_cost
)
else
:
label_probs
=
tf
.
nn
.
softmax
(
fastrcnn_label_logits
,
name
=
'fastrcnn_all_probs'
)
# #proposal x #Class
anchors
=
tf
.
tile
(
tf
.
expand_dims
(
proposal_boxes
,
1
),
[
1
,
config
.
NUM_CLASS
-
1
,
1
])
# #proposal x #Cat x 4
...
...
examples/README.md
View file @
8bc909be
# tensorpack examples
Training examples with __reproducible__ and meaningful performance.
Training examples with __reproducible__ performance.
__Reproducible performance is important__
. Usually deep learning code is easy to write,
but hard to know the correctness -- wrong code will usually still converge.
Without a setting and performance comparable to someone else, you don't know if an implementation is correct or not.
## Getting Started:
+
[
An illustrative mnist example with explanation of the framework
](
mnist-convnet.py
)
...
...
@@ -13,7 +17,7 @@ Training examples with __reproducible__ and meaningful performance.
| Name | Performance |
| --- | --- |
| Train
[
ResNet
](
ResNet
)
and
[
ShuffleNet
](
ShuffleNet
)
on ImageNet | reproduce paper |
|
[
Train
ResNet50-
Faster-RCNN on COCO
](
FasterRCNN
)
| reproduce paper |
|
[
Train Faster-RCNN on COCO
](
FasterRCNN
)
| reproduce paper |
|
[
DoReFa-Net: training binary / low-bitwidth CNN on ImageNet
](
DoReFa-Net
)
| reproduce paper |
|
[
Generative Adversarial Network(GAN) variants
](
GAN
)
, including DCGAN, InfoGAN,
<br/>
Conditional GAN, WGAN, BEGAN, DiscoGAN, Image to Image, CycleGAN | visually reproduce |
|
[
Inception-BN and InceptionV3
](
Inception
)
| reproduce reference code |
...
...
@@ -44,7 +48,3 @@ Training examples with __reproducible__ and meaningful performance.
Example needs to satisfy one of the following:
+
Reproduce performance of a published or well-known paper.
+
Illustrate a new way of using the library that is currently not covered.
__Performance is important__
. Usually deep learning code is easy to write,
but hard to know the correctness -- thanks to SGD things will usually still converge when you've made mistakes.
Without a setting and performance comparable to someone else, you don't know if an implementation is correct or not.
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