wehub-resource-sync
67 lines
3.5 KiB
Python
67 lines
3.5 KiB
Python
"""Loads the tflite_mobilenet libraries, used by TF."""
|
|
|
|
load("//third_party:repo.bzl", "tf_http_archive")
|
|
|
|
def repo():
|
|
"""Loads the tflite_mobilenet libraries, used by TF."""
|
|
tf_http_archive(
|
|
name = "tflite_mobilenet_float",
|
|
build_file = "//third_party/tflite_mobilenet:tflite_mobilenet_float.BUILD",
|
|
sha256 = "2fadeabb9968ec6833bee903900dda6e61b3947200535874ce2fe42a8493abc0",
|
|
urls = [
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224.tgz",
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224.tgz",
|
|
],
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "tflite_mobilenet_quant",
|
|
build_file = "//third_party/tflite_mobilenet:tflite_mobilenet_quant.BUILD",
|
|
sha256 = "d32432d28673a936b2d6281ab0600c71cf7226dfe4cdcef3012555f691744166",
|
|
urls = [
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224_quant.tgz",
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/mobilenet_v1_2018_08_02/mobilenet_v1_1.0_224_quant.tgz",
|
|
],
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "tflite_mobilenet_ssd",
|
|
build_file = "//third_party/tflite_mobilenet:tflite_mobilenet.BUILD",
|
|
sha256 = "767057f2837a46d97882734b03428e8dd640b93236052b312b2f0e45613c1cf0",
|
|
urls = [
|
|
"https://storage.googleapis.com/mirror.tensorflow.org/storage.googleapis.com/download.tensorflow.org/models/tflite/mobilenet_ssd_tflite_v1.zip",
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/tflite/mobilenet_ssd_tflite_v1.zip",
|
|
],
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "tflite_mobilenet_ssd_quant",
|
|
build_file = "//third_party/tflite_mobilenet:tflite_mobilenet.BUILD",
|
|
sha256 = "a809cd290b4d6a2e8a9d5dad076e0bd695b8091974e0eed1052b480b2f21b6dc",
|
|
urls = [
|
|
"https://storage.googleapis.com/mirror.tensorflow.org/storage.googleapis.com/download.tensorflow.org/models/tflite/coco_ssd_mobilenet_v1_0.75_quant_2018_06_29.zip",
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/tflite/coco_ssd_mobilenet_v1_0.75_quant_2018_06_29.zip",
|
|
],
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "tflite_mobilenet_ssd_quant_protobuf",
|
|
build_file = "//third_party/tflite_mobilenet:tflite_mobilenet.BUILD",
|
|
sha256 = "09280972c5777f1aa775ef67cb4ac5d5ed21970acd8535aeca62450ef14f0d79",
|
|
strip_prefix = "ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18",
|
|
urls = [
|
|
"https://storage.googleapis.com/mirror.tensorflow.org/storage.googleapis.com/download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18.tar.gz",
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/object_detection/ssd_mobilenet_v1_quantized_300x300_coco14_sync_2018_07_18.tar.gz",
|
|
],
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "tflite_conv_actions_frozen",
|
|
build_file = "//third_party/tflite_mobilenet:tflite_mobilenet.BUILD",
|
|
sha256 = "d947b38cba389b5e2d0bfc3ea6cc49c784e187b41a071387b3742d1acac7691e",
|
|
urls = [
|
|
"https://storage.googleapis.com/mirror.tensorflow.org/storage.googleapis.com/download.tensorflow.org/models/tflite/conv_actions_tflite.zip",
|
|
"https://storage.googleapis.com/download.tensorflow.org/models/tflite/conv_actions_tflite.zip",
|
|
],
|
|
)
|