Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
ResNet50_tensorflow
Commits
975fbbde
Commit
975fbbde
authored
Jun 15, 2021
by
Vishnu Banna
Browse files
new lines
parent
fd50cc1b
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
15 additions
and
13 deletions
+15
-13
official/vision/beta/projects/yolo/README.md
official/vision/beta/projects/yolo/README.md
+14
-12
official/vision/beta/projects/yolo/modeling/layers/nn_blocks.py
...al/vision/beta/projects/yolo/modeling/layers/nn_blocks.py
+1
-0
official/vision/beta/projects/yolo/modeling/layers/nn_blocks_test.py
...sion/beta/projects/yolo/modeling/layers/nn_blocks_test.py
+0
-1
No files found.
official/vision/beta/projects/yolo/README.md
View file @
975fbbde
...
...
@@ -17,9 +17,9 @@ repository.
## Description
YOLO v1 the original implementation was released in 2015 providing a
groundbreaking
algorithm that would quickly process images and locate objects in
a
single pass through the detector. The original implementation used a
YOLO v1 the original implementation was released in 2015 providing a
groundbreaking
algorithm that would quickly process images and locate objects in
a
single pass through the detector. The original implementation used a
backbone derived from state of the art object classifiers of the time, like
[
GoogLeNet
](
https://arxiv.org/abs/1409.4842
)
and
[
VGG
](
https://arxiv.org/abs/1409.1556
)
. More attention was given to the novel
...
...
@@ -32,14 +32,15 @@ update and develop this model.
YOLO v3 and v4 serve as the most up to date and capable versions of the YOLO
network group. This model uses a custom backbone called Darknet53 that uses
knowledge gained from the ResNet paper to improve its predictions. The new backbone
also allows for objects to be detected at multiple scales. As for the new detection head,
the model now predicts the bounding boxes using a set of anchor box priors (Anchor
Boxes) as suggestions. Multiscale predictions in combination with Anchor boxes allow
for the network to make up to 1000 object predictions on a single image. Finally,
the new loss function forces the network to make better predictions by using Intersection
Over Union (IOU) to inform the model's confidence rather than relying on the mean
squared error for the entire output.
knowledge gained from the ResNet paper to improve its predictions. The new
backbone also allows for objects to be detected at multiple scales. As for the
new detection head, the model now predicts the bounding boxes using a set of
anchor box priors (Anchor Boxes) as suggestions. Multiscale predictions in
combination with Anchor boxes allow for the network to make up to 1000 object
predictions on a single image. Finally, the new loss function forces the network
to make better predictions by using Intersection Over Union (IOU) to inform the
model's confidence rather than relying on the mean squared error for the entire
output.
## Authors
...
...
@@ -79,4 +80,5 @@ connected to a new, more powerful backbone if a person chose to.
[

](https://www.python.org/downloads/release/python-380/)
DISCLAIMER: this YOLO implementation is still under development. No support will be provided during the development phase.
DISCLAIMER: this YOLO implementation is still under development. No support
will be provided during the development phase.
official/vision/beta/projects/yolo/modeling/layers/nn_blocks.py
View file @
975fbbde
...
...
@@ -1360,6 +1360,7 @@ class DarkRouteProcess(tf.keras.layers.Layer):
repetitions = 3,
insert_spp = False)(x)
"""
def
__init__
(
self
,
filters
=
2
,
...
...
official/vision/beta/projects/yolo/modeling/layers/nn_blocks_test.py
View file @
975fbbde
...
...
@@ -62,7 +62,6 @@ class CSPConnectTest(tf.test.TestCase, parameterized.TestCase):
grad_loss
=
loss
(
x_hat
,
y
)
grad
=
tape
.
gradient
(
grad_loss
,
test_layer
.
trainable_variables
)
optimizer
.
apply_gradients
(
zip
(
grad
,
test_layer
.
trainable_variables
))
self
.
assertNotIn
(
None
,
grad
)
...
...
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