From 505bb5bad6f559a5d151fac36abb09a8f25a1a66 Mon Sep 17 00:00:00 2001 From: pfs-15 Date: Fri, 20 Dec 2019 15:40:06 +0100 Subject: [PATCH] Add 5 additional model presets --- orx-runway/src/main/kotlin/Presets.kt | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) diff --git a/orx-runway/src/main/kotlin/Presets.kt b/orx-runway/src/main/kotlin/Presets.kt index 34681b56..5b705e57 100644 --- a/orx-runway/src/main/kotlin/Presets.kt +++ b/orx-runway/src/main/kotlin/Presets.kt @@ -2,8 +2,30 @@ package org.openrndr.extra.runway import com.google.gson.annotations.SerializedName +// -- AttnGAN class CaptionRequest(val caption: String) class CaptionResult(val result: String) +// -- BDCN +class BdcnRequest(val input_image: String) +class BdcnResult(val output_image: String) + +// -- BigBiGAN class BigBiGANQuery(@SerializedName("input_image") val inputImage: String) class BigBiGANResult(@SerializedName("output_image") val outputImage: String) + +// -- SPADE-COCO +class SpadeCocoRequest(val semantic_map: String) +class SpadeCocoResult(val output: String) + +// -- GPT-2 +class Gpt2Request(val prompt: String) +class Gpt2Result(val text: String) + +// -- im2txt +class Im2txtRequest(val image: String) +class Im2txtResult(val caption: String) + +// -- PSENet +class PsenetRequest(@SerializedName("input_image") val inputImage: String) +class PsenetResult(val bboxes: Array>) \ No newline at end of file