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
02f5599f
Commit
02f5599f
authored
Jul 06, 2022
by
ghpark
Browse files
Internal change
parent
8b47f2c6
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
1 addition
and
4 deletions
+1
-4
official/projects/detr/dataloaders/detr_input.py
official/projects/detr/dataloaders/detr_input.py
+1
-3
official/projects/detr/modeling/detr.py
official/projects/detr/modeling/detr.py
+0
-1
No files found.
official/projects/detr/dataloaders/detr_input.py
View file @
02f5599f
...
...
@@ -14,15 +14,13 @@
"""COCO data loader for DETR."""
from
typing
import
Optional
,
Tuple
from
typing
import
Tuple
import
tensorflow
as
tf
from
official.vision.dataloaders
import
parser
from
official.vision.dataloaders
import
utils
from
official.vision.ops
import
box_ops
from
official.vision.ops
import
preprocess_ops
from
official.core
import
input_reader
RESIZE_SCALES
=
(
480
,
512
,
544
,
576
,
608
,
640
,
672
,
704
,
736
,
768
,
800
)
...
...
official/projects/detr/modeling/detr.py
View file @
02f5599f
...
...
@@ -24,7 +24,6 @@ import tensorflow as tf
from
official.modeling
import
tf_utils
from
official.projects.detr.modeling
import
transformer
from
official.vision.modeling.backbones
import
resnet
def
position_embedding_sine
(
attention_mask
,
...
...
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