Add the model configuration and training hyperparameters for BlazeFaceStylizer.

PiperOrigin-RevId: 520767282
This commit is contained in:
MediaPipe Team
2023-03-30 16:07:11 -07:00
committed by Copybara-Service
parent a4923ca7aa
commit 1eeb89e95f
6 changed files with 253 additions and 0 deletions
@@ -26,6 +26,40 @@ filegroup(
]),
)
py_library(
name = "hyperparameters",
srcs = ["hyperparameters.py"],
deps = [
"//mediapipe/model_maker/python/core:hyperparameters",
],
)
py_library(
name = "model_options",
srcs = ["model_options.py"],
)
py_library(
name = "model_spec",
srcs = ["model_spec.py"],
)
py_test(
name = "model_spec_test",
srcs = ["model_spec_test.py"],
deps = [":model_spec"],
)
py_library(
name = "face_stylizer_options",
srcs = ["face_stylizer_options.py"],
deps = [
":hyperparameters",
":model_options",
":model_spec",
],
)
py_library(
name = "dataset",
srcs = ["dataset.py"],