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.
35 lines
887 B
35 lines
887 B
/**
|
|
* To download necessary library from HuaWei server.
|
|
* Including mindspore-lite .so file, minddata-lite .so file and model file.
|
|
* The libraries can be downloaded manually.
|
|
*/
|
|
def targetModelFile = "src/main/assets/posenet_model.ms"
|
|
def modelDownloadUrl = "https://download.mindspore.cn/model_zoo/official/lite/posenet_lite/posenet_model.ms"
|
|
|
|
task downloadModelFile(type: DownloadUrlTask) {
|
|
doFirst {
|
|
println "Downloading ${modelDownloadUrl}"
|
|
}
|
|
sourceUrl = "${modelDownloadUrl}"
|
|
target = file("${targetModelFile}")
|
|
}
|
|
|
|
if (file("src/main/assets/posenet_model.ms").exists()) {
|
|
downloadModelFile.enabled = false
|
|
}
|
|
|
|
preBuild.dependsOn downloadModelFile
|
|
|
|
class DownloadUrlTask extends DefaultTask {
|
|
@Input
|
|
String sourceUrl
|
|
|
|
@OutputFile
|
|
File target
|
|
|
|
@TaskAction
|
|
void download() {
|
|
ant.get(src: sourceUrl, dest: target)
|
|
}
|
|
}
|