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
chenpangpang
transformers
Commits
cb0c6806
Unverified
Commit
cb0c6806
authored
Oct 24, 2023
by
amyeroberts
Committed by
GitHub
Oct 24, 2023
Browse files
Safe import of rgb_to_id from FE modules (#27037)
Safe import from FE modules
parent
7bde5d63
Changes
4
Show whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
40 additions
and
0 deletions
+40
-0
src/transformers/models/conditional_detr/feature_extraction_conditional_detr.py
...s/conditional_detr/feature_extraction_conditional_detr.py
+10
-0
src/transformers/models/deformable_detr/feature_extraction_deformable_detr.py
...els/deformable_detr/feature_extraction_deformable_detr.py
+10
-0
src/transformers/models/detr/feature_extraction_detr.py
src/transformers/models/detr/feature_extraction_detr.py
+10
-0
src/transformers/models/yolos/feature_extraction_yolos.py
src/transformers/models/yolos/feature_extraction_yolos.py
+10
-0
No files found.
src/transformers/models/conditional_detr/feature_extraction_conditional_detr.py
View file @
cb0c6806
...
...
@@ -16,6 +16,7 @@
import
warnings
from
...image_transforms
import
rgb_to_id
as
_rgb_to_id
from
...utils
import
logging
from
.image_processing_conditional_detr
import
ConditionalDetrImageProcessor
...
...
@@ -23,6 +24,15 @@ from .image_processing_conditional_detr import ConditionalDetrImageProcessor
logger
=
logging
.
get_logger
(
__name__
)
def
rgb_to_id
(
x
):
warnings
.
warn
(
"rgb_to_id has moved and will not be importable from this module from v5. "
"Please import from transformers.image_transforms instead."
,
FutureWarning
,
)
return
_rgb_to_id
(
x
)
class
ConditionalDetrFeatureExtractor
(
ConditionalDetrImageProcessor
):
def
__init__
(
self
,
*
args
,
**
kwargs
)
->
None
:
warnings
.
warn
(
...
...
src/transformers/models/deformable_detr/feature_extraction_deformable_detr.py
View file @
cb0c6806
...
...
@@ -16,6 +16,7 @@
import
warnings
from
...image_transforms
import
rgb_to_id
as
_rgb_to_id
from
...utils
import
logging
from
.image_processing_deformable_detr
import
DeformableDetrImageProcessor
...
...
@@ -23,6 +24,15 @@ from .image_processing_deformable_detr import DeformableDetrImageProcessor
logger
=
logging
.
get_logger
(
__name__
)
def
rgb_to_id
(
x
):
warnings
.
warn
(
"rgb_to_id has moved and will not be importable from this module from v5. "
"Please import from transformers.image_transforms instead."
,
FutureWarning
,
)
return
_rgb_to_id
(
x
)
class
DeformableDetrFeatureExtractor
(
DeformableDetrImageProcessor
):
def
__init__
(
self
,
*
args
,
**
kwargs
)
->
None
:
warnings
.
warn
(
...
...
src/transformers/models/detr/feature_extraction_detr.py
View file @
cb0c6806
...
...
@@ -16,6 +16,7 @@
import
warnings
from
...image_transforms
import
rgb_to_id
as
_rgb_to_id
from
...utils
import
logging
from
.image_processing_detr
import
DetrImageProcessor
...
...
@@ -23,6 +24,15 @@ from .image_processing_detr import DetrImageProcessor
logger
=
logging
.
get_logger
(
__name__
)
def
rgb_to_id
(
x
):
warnings
.
warn
(
"rgb_to_id has moved and will not be importable from this module from v5. "
"Please import from transformers.image_transforms instead."
,
FutureWarning
,
)
return
_rgb_to_id
(
x
)
class
DetrFeatureExtractor
(
DetrImageProcessor
):
def
__init__
(
self
,
*
args
,
**
kwargs
)
->
None
:
warnings
.
warn
(
...
...
src/transformers/models/yolos/feature_extraction_yolos.py
View file @
cb0c6806
...
...
@@ -16,6 +16,7 @@
import
warnings
from
...image_transforms
import
rgb_to_id
as
_rgb_to_id
from
...utils
import
logging
from
.image_processing_yolos
import
YolosImageProcessor
...
...
@@ -23,6 +24,15 @@ from .image_processing_yolos import YolosImageProcessor
logger
=
logging
.
get_logger
(
__name__
)
def
rgb_to_id
(
x
):
warnings
.
warn
(
"rgb_to_id has moved and will not be importable from this module from v5. "
"Please import from transformers.image_transforms instead."
,
FutureWarning
,
)
return
_rgb_to_id
(
x
)
class
YolosFeatureExtractor
(
YolosImageProcessor
):
def
__init__
(
self
,
*
args
,
**
kwargs
)
->
None
:
warnings
.
warn
(
...
...
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