diff --git a/build.gradle.kts b/build.gradle.kts index 6b3f24a..22657dc 100644 --- a/build.gradle.kts +++ b/build.gradle.kts @@ -47,10 +47,24 @@ val orxFeatures = setOf( // "orx-temporal-blur", // "orx-time-operators", // "orx-kinect-v1", - +// "orx-tensorflow", + "orx-panel" ) +val ormlFeatures = setOf( +// "orml-blazepose", +// "orml-dbface", +// "orml-facemesh", +// "orml-image-classifier", +// "orml-psenet", +// "orml-ssd", +// "orml-style-transfer", +// "orml-super-resolution", +// "orml-u2net" +) + + /* Which OPENRNDR libraries should be added to this project? */ val openrndrFeatures = setOf( "video" @@ -63,6 +77,12 @@ val openrndrVersion = if (openrndrUseSnapshot) "0.4.0-SNAPSHOT" else "0.3.44" val orxUseSnapshot = false val orxVersion = if (orxUseSnapshot) "0.4.0-SNAPSHOT" else "0.3.53" +val ormlUseSnapshot = false +val ormlVersion = if (ormlUseSnapshot) "0.4.0-SNAPSHOT" else "0.3.0-rc.5" + +// choices are "orx-tensorflow-gpu", "orx-tensorflow-mkl", "orx-tensorflow" +val orxTensorflowBackend = "orx-tensorflow-mkl" + // val supportedPlatforms = setOf("windows", "macos", "linux-x64", "linux-arm64") @@ -115,6 +135,10 @@ fun DependencyHandler.orx(module: String): Any { return "org.openrndr.extra:$module:$orxVersion" } +fun DependencyHandler.orml(module: String): Any { + return "org.openrndr.orml:$module:$ormlVersion" +} + fun DependencyHandler.openrndr(module: String): Any { return "org.openrndr:openrndr-$module:$openrndrVersion" } @@ -168,6 +192,14 @@ dependencies { for (feature in orxFeatures) { implementation(orx(feature)) } + + for (feature in ormlFeatures) { + implementation(orml(feature)) + } + + if ("orx-tensorflow" in orxFeatures) { + runtimeOnly("org.openrndr.extra:$orxTensorflowBackend-natives-$openrndrOs:$orxVersion") + } if ("orx-kinect-v1" in orxFeatures) { runtimeOnly(orxNatives("orx-kinect-v1"))