You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.4 KiB
68 lines
2.4 KiB
task downloadPosenetModel(type: DownloadUrlTask) {
|
|
def modelPosenetDownloadUrl = "https://storage.googleapis.com/download.tensorflow.org/models/tflite/posenet_mobilenet_v1_100_257x257_multi_kpt_stripped.tflite"
|
|
doFirst {
|
|
println "Downloading ${modelPosenetDownloadUrl}"
|
|
}
|
|
sourceUrl = "${modelPosenetDownloadUrl}"
|
|
target = file("src/main/assets/posenet.tflite")
|
|
}
|
|
|
|
task downloadMovenetLightningModel(type: DownloadUrlTask) {
|
|
def modelMovenetLightningDownloadUrl = "https://tfhub.dev/google/lite-model/movenet/singlepose/lightning/tflite/float16/4?lite-format=tflite"
|
|
doFirst {
|
|
println "Downloading ${modelMovenetLightningDownloadUrl}"
|
|
}
|
|
sourceUrl = "${modelMovenetLightningDownloadUrl}"
|
|
target = file("src/main/assets/movenet_lightning.tflite")
|
|
}
|
|
|
|
task downloadMovenetThunderModel(type: DownloadUrlTask) {
|
|
def modelMovenetThunderDownloadUrl = "https://tfhub.dev/google/lite-model/movenet/singlepose/thunder/tflite/float16/4?lite-format=tflite"
|
|
doFirst {
|
|
println "Downloading ${modelMovenetThunderDownloadUrl}"
|
|
}
|
|
sourceUrl = "${modelMovenetThunderDownloadUrl}"
|
|
target = file("src/main/assets/movenet_thunder.tflite")
|
|
}
|
|
|
|
task downloadMovenetMultiPoseModel(type: DownloadUrlTask) {
|
|
def modelMovenetThunderDownloadUrl = "https://tfhub.dev/google/lite-model/movenet/multipose/lightning/tflite/float16/1?lite-format=tflite"
|
|
doFirst {
|
|
println "Downloading ${modelMovenetThunderDownloadUrl}"
|
|
}
|
|
sourceUrl = "${modelMovenetThunderDownloadUrl}"
|
|
target = file("src/main/assets/movenet_multipose_fp16.tflite")
|
|
}
|
|
|
|
task downloadPoseClassifierModel(type: DownloadUrlTask) {
|
|
def modelPoseClassifierDownloadUrl = "https://storage.googleapis.com/download.tensorflow.org/models/tflite/pose_classifier/yoga_classifier.tflite"
|
|
doFirst {
|
|
println "Downloading ${modelPoseClassifierDownloadUrl}"
|
|
}
|
|
sourceUrl = "${modelPoseClassifierDownloadUrl}"
|
|
target = file("src/main/assets/classifier.tflite")
|
|
}
|
|
|
|
task downloadModel {
|
|
dependsOn downloadPosenetModel
|
|
dependsOn downloadMovenetLightningModel
|
|
dependsOn downloadMovenetThunderModel
|
|
dependsOn downloadPoseClassifierModel
|
|
dependsOn downloadMovenetMultiPoseModel
|
|
}
|
|
|
|
class DownloadUrlTask extends DefaultTask {
|
|
@Input
|
|
String sourceUrl
|
|
|
|
@OutputFile
|
|
File target
|
|
|
|
@TaskAction
|
|
void download() {
|
|
ant.get(src: sourceUrl, dest: target)
|
|
}
|
|
}
|
|
|
|
preBuild.dependsOn downloadModel
|