Skip to content

Adding RegNets to tf.keras.applications #15702

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 12 commits into from
Closed

Adding RegNets to tf.keras.applications #15702

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
275565f
Restored progress from #15419
AdityaKane2001 Nov 25, 2021
6c719b1
Used f-strings for block numbering in Stage
AdityaKane2001 Nov 27, 2021
fb05277
Added tf_py_test, docstrings and other minor corrections
AdityaKane2001 Dec 1, 2021
892813f
Added RegNetX sha256 hashes
AdityaKane2001 Dec 7, 2021
7297820
Added all RegNetY sha256 hashes except Y004
AdityaKane2001 Dec 8, 2021
98d67e5
Added sha256 hashes for Y004
AdityaKane2001 Dec 10, 2021
4383429
Minor error in BASE_DOCSTRING
AdityaKane2001 Dec 10, 2021
c7f2151
Change in applications tset for grouped conv compatibility
AdityaKane2001 Dec 10, 2021
e17f378
Minor changes and yapf
AdityaKane2001 Dec 16, 2021
851ca16
Compat change in applications_load_weight_test
AdityaKane2001 Dec 24, 2021
cf25748
Rollback 851ca16
AdityaKane2001 Jan 9, 2022
62c79eb
Rollback 851ca16
AdityaKane2001 Jan 9, 2022
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions keras/api/BUILD
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ keras_packages = [
"keras.applications.mobilenet_v2",
"keras.applications.mobilenet_v3",
"keras.applications.nasnet",
"keras.applications.regnet",
"keras.applications.resnet",
"keras.applications.resnet_v2",
"keras.applications.vgg16",
Expand Down
2 changes: 2 additions & 0 deletions keras/api/api_init_files.bzl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -21,6 +21,7 @@ KERAS_API_INIT_FILES = [
"keras/applications/mobilenet_v2/__init__.py",
"keras/applications/mobilenet_v3/__init__.py",
"keras/applications/nasnet/__init__.py",
"keras/applications/regnet/__init__.py",
"keras/applications/resnet/__init__.py",
"keras/applications/resnet50/__init__.py",
"keras/applications/resnet_v2/__init__.py",
Expand Down Expand Up @@ -85,6 +86,7 @@ KERAS_API_INIT_FILES_V1 = [
"keras/applications/mobilenet_v2/__init__.py",
"keras/applications/mobilenet_v3/__init__.py",
"keras/applications/nasnet/__init__.py",
"keras/applications/regnet/__init__.py",
"keras/applications/resnet/__init__.py",
"keras/applications/resnet_v2/__init__.py",
"keras/applications/resnet50/__init__.py",
Expand Down
18 changes: 18 additions & 0 deletions keras/applications/BUILD
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@ py_library(
"mobilenet_v2.py",
"mobilenet_v3.py",
"nasnet.py",
"regnet.py",
"resnet.py",
"resnet_v2.py",
"vgg16.py",
Expand Down Expand Up @@ -312,6 +313,23 @@ tf_py_test(
],
)

tf_py_test(
name = "applications_load_weight_test_regnet",
srcs = ["applications_load_weight_test.py"],
args = ["--module=regnet"],
main = "applications_load_weight_test.py",
tags = [
"no_oss",
"no_pip",
],
deps = [
":applications",
"//:expect_absl_installed",
"//:expect_tensorflow_installed",
"//keras/preprocessing",
],
)

tf_py_test(
name = "applications_load_weight_test_nasnet_mobile",
srcs = ["applications_load_weight_test.py"],
Expand Down
12 changes: 11 additions & 1 deletion keras/applications/applications_load_weight_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -29,6 +29,7 @@
from keras.applications import mobilenet_v2
from keras.applications import mobilenet_v3
from keras.applications import nasnet
from keras.applications import regnet
from keras.applications import resnet
from keras.applications import resnet_v2
from keras.applications import vgg16
Expand Down Expand Up @@ -69,7 +70,16 @@
efficientnet_v2.EfficientNetV2B2, efficientnet_v2.EfficientNetV2B3,
efficientnet_v2.EfficientNetV2S, efficientnet_v2.EfficientNetV2M,
efficientnet_v2.EfficientNetV2L
])
]),
'regnet': (regnet,
[regnet.RegNetX002, regnet.RegNetX004, regnet.RegNetX006,
regnet.RegNetX008, regnet.RegNetX016, regnet.RegNetX032,
regnet.RegNetX040, regnet.RegNetX064, regnet.RegNetX080,
regnet.RegNetX120, regnet.RegNetX160, regnet.RegNetX320,
regnet.RegNetY002, regnet.RegNetY004, regnet.RegNetY006,
regnet.RegNetY008, regnet.RegNetY016, regnet.RegNetY032,
regnet.RegNetY040, regnet.RegNetY064, regnet.RegNetY080,
regnet.RegNetY120, regnet.RegNetY160, regnet.RegNetY320])
}

TEST_IMAGE_PATH = ('https://storage.googleapis.com/tensorflow/'
Expand Down
25 changes: 25 additions & 0 deletions keras/applications/applications_test.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
from keras.applications import mobilenet_v2
from keras.applications import mobilenet_v3
from keras.applications import nasnet
from keras.applications import regnet
from keras.applications import resnet
from keras.applications import resnet_v2
from keras.applications import vgg16
Expand Down Expand Up @@ -69,6 +70,30 @@
(efficientnet_v2.EfficientNetV2S, 1280),
(efficientnet_v2.EfficientNetV2M, 1280),
(efficientnet_v2.EfficientNetV2L, 1280),
(regnet.RegNetX002, 368),
(regnet.RegNetX004, 384),
(regnet.RegNetX006, 528),
(regnet.RegNetX008, 672),
(regnet.RegNetX016, 912),
(regnet.RegNetX032, 1008),
(regnet.RegNetX040, 1360),
(regnet.RegNetX064, 1624),
(regnet.RegNetX080, 1920),
(regnet.RegNetX120, 2240),
(regnet.RegNetX160, 2048),
(regnet.RegNetX320, 2520),
(regnet.RegNetY002, 368),
(regnet.RegNetY004, 440),
(regnet.RegNetY006, 608),
(regnet.RegNetY008, 768),
(regnet.RegNetY016, 888),
(regnet.RegNetY032, 1512),
(regnet.RegNetY040, 1088),
(regnet.RegNetY064, 1296),
(regnet.RegNetY080, 2016),
(regnet.RegNetY120, 2240),
(regnet.RegNetY160, 3024),
(regnet.RegNetY320, 3712)
]

NASNET_LIST = [
Expand Down
Loading