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
7ecd20ae
Commit
7ecd20ae
authored
Mar 02, 2022
by
Yeqing Li
Committed by
A. Unique TensorFlower
Mar 02, 2022
Browse files
Internal change
PiperOrigin-RevId: 432072951
parent
10f8f1e1
Changes
3
Show whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
5 additions
and
5 deletions
+5
-5
official/projects/detr/dataloaders/coco.py
official/projects/detr/dataloaders/coco.py
+2
-2
official/projects/detr/modeling/detr.py
official/projects/detr/modeling/detr.py
+1
-1
official/projects/detr/tasks/detection.py
official/projects/detr/tasks/detection.py
+2
-2
No files found.
official/projects/detr/dataloaders/coco.py
View file @
7ecd20ae
...
@@ -20,8 +20,8 @@ import tensorflow as tf
...
@@ -20,8 +20,8 @@ import tensorflow as tf
from
official.core
import
config_definitions
as
cfg
from
official.core
import
config_definitions
as
cfg
from
official.core
import
input_reader
from
official.core
import
input_reader
from
official.vision.
beta.
ops
import
box_ops
from
official.vision.ops
import
box_ops
from
official.vision.
beta.
ops
import
preprocess_ops
from
official.vision.ops
import
preprocess_ops
@
dataclasses
.
dataclass
@
dataclasses
.
dataclass
...
...
official/projects/detr/modeling/detr.py
View file @
7ecd20ae
...
@@ -24,7 +24,7 @@ import tensorflow as tf
...
@@ -24,7 +24,7 @@ import tensorflow as tf
from
official.modeling
import
tf_utils
from
official.modeling
import
tf_utils
from
official.projects.detr.modeling
import
transformer
from
official.projects.detr.modeling
import
transformer
from
official.vision.
beta.
modeling.backbones
import
resnet
from
official.vision.modeling.backbones
import
resnet
def
position_embedding_sine
(
attention_mask
,
def
position_embedding_sine
(
attention_mask
,
...
...
official/projects/detr/tasks/detection.py
View file @
7ecd20ae
...
@@ -22,8 +22,8 @@ from official.projects.detr.configs import detr as detr_cfg
...
@@ -22,8 +22,8 @@ from official.projects.detr.configs import detr as detr_cfg
from
official.projects.detr.dataloaders
import
coco
from
official.projects.detr.dataloaders
import
coco
from
official.projects.detr.modeling
import
detr
from
official.projects.detr.modeling
import
detr
from
official.projects.detr.ops
import
matchers
from
official.projects.detr.ops
import
matchers
from
official.vision.
beta.
evaluation
import
coco_evaluator
from
official.vision.evaluation
import
coco_evaluator
from
official.vision.
beta.
ops
import
box_ops
from
official.vision.ops
import
box_ops
@
task_factory
.
register_task_cls
(
detr_cfg
.
DetectionConfig
)
@
task_factory
.
register_task_cls
(
detr_cfg
.
DetectionConfig
)
...
...
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