kaze/orx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #19 from pfs-15/master

Browse Source 
Add 5 additional model presets
This commit is contained in:
Edwin Jakobs
2019-12-20 16:46:08 +01:00
committed by GitHub
parent 01a39d4591 505bb5bad6
commit 18bda2e364
1 changed files with 22 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

22
orx-runway/src/main/kotlin/Presets.kt
View File

@@ -2,8 +2,30 @@ package org.openrndr.extra.runway
import com.google.gson.annotations.SerializedName import com.google.gson.annotations.SerializedName
// -- AttnGAN
class CaptionRequest(val caption: String) class CaptionRequest(val caption: String)
class CaptionResult(val result: 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 BigBiGANQuery(@SerializedName("input_image") val inputImage: String)
class BigBiGANResult(@SerializedName("output_image") val outputImage: 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<Array<Double>>)