mixvideo-v2/cargos/tvai-v2/models/rhea-1.json

575 lines
16 KiB
JSON

{
"autoModel": "prap-3",
"backends": {
"coreml": {
"model": "rhea",
"parallel": 1,
"scales": {
"1": {
"spec": {
"13": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"devices": [
-2
],
"model": "rhea",
"nets": [
"fgnet-fp16-[H]x[W]-4x-13-mlp.tz"
],
"outputs": [
"model_output"
],
"parallel": 1
}
}
},
"2": {
"spec": {
"13": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"devices": [
-2
],
"model": "rhea",
"nets": [
"fgnet-fp16-[H]x[W]-4x-13-mlp.tz"
],
"outputs": [
"model_output"
],
"parallel": 1
}
}
},
"4": {
"spec": {
"13": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"devices": [
-2
],
"model": "rhea",
"nets": [
"fgnet-fp16-[H]x[W]-[S]x-13-mlp.tz"
],
"outputs": [
"model_output"
],
"parallel": 1
}
}
}
}
},
"onnx": {
"model": "rhea",
"parallel": 1,
"scales": {
"1": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp32-[H]x[W]-4x-ox.tz"
],
"outputs": [
"out"
]
},
"2": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp32-[H]x[W]-4x-ox.tz"
],
"outputs": [
"out"
]
},
"4": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp32-[H]x[W]-[S]x-ox.tz"
],
"outputs": [
"out"
]
}
}
},
"onnx16": {
"model": "rhea",
"parallel": 1,
"scales": {
"1": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp16-[H]x[W]-4x-ox.tz"
],
"outputs": [
"out"
]
},
"2": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp16-[H]x[W]-4x-ox.tz"
],
"outputs": [
"out"
]
},
"4": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp16-[H]x[W]-[S]x-ox.tz"
],
"outputs": [
"out"
]
}
}
},
"openvino": {
"model": "rhea",
"parallel": 1,
"scales": {
"1": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp16-[H]x[W]-4x-ox.tz"
],
"outputs": [
"out"
]
},
"2": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp16-[H]x[W]-4x-ox.tz"
],
"outputs": [
"out"
]
},
"4": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp16-[H]x[W]-[S]x-ox.tz"
],
"outputs": [
"out"
]
}
}
},
"tensorrt": {
"capabilities": [
1200,
809,
806,
705
],
"model": "rhea",
"parallel": 1,
"scales": {
"1": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp32-[H]x[W]-4x-rt[C]-[R]-rt.tz"
],
"outputs": [
"out"
]
},
"2": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp32-[H]x[W]-4x-rt[C]-[R]-rt.tz"
],
"outputs": [
"out"
]
},
"4": {
"blocks": [
288,
288,
352,
256,
480,
384,
384,
480,
384,
576,
416,
576,
576,
448,
672,
384,
672,
576
],
"nets": [
"fgnet-fp32-[H]x[W]-[S]x-rt[C]-[R]-rt.tz"
],
"outputs": [
"out"
]
}
}
}
},
"dependencies": [
"prap-3"
],
"displayName": "Rhea",
"enabled": 1,
"gui": {
"name": "Rhea - Medium Quality",
"version": "1"
},
"inputs": {
"hr_prev": "hr_prev",
"lr_curr": "lr_curr",
"lr_next": "lr_next",
"lr_prev": "lr_prev",
"params": "params"
},
"frames": 3,
"interlacedFrames": 0,
"jsonVersion": "1.0.0",
"keepColor": 1,
"key": "CHJPPIOMOMCPNLKEOLFNAPEBPGDDEIGBOOIDFPNMKAMDOEEFMIJJOOEFMEGNMAJJGIGJEPNJKCLDBKJIONNHGKDLPMAOGCHBGFBIHKGHACOBNLOBBONCFFCLBLDIAJNJCDBGFGABNBPJFEFONMKGPEPLBPPKEEGPBAGBEHADIECMMBIBPBGLBLACMMAOFABHLLECPNFGBBFONGNLPBIOILMGKJHDBNKFNIGBIEPMENLMCOLDOEMEPKOBBGJILJBFKIHNMPAIPAIAGDLINMODHEBPCFOOOMDFCECFHGLAKFCLJGHGLMGLLFPKJGOJHENF",
"minAppVersion": "5.0.0",
"minCMLMacVersion": 13.0,
"modelType": 1,
"noiseLevel": 0.4,
"parameters": [
{
"default": 0,
"max": 1,
"min": -1,
"name": "preBlur",
"guiName": "Anti-Alias/Deblur"
},
{
"default": 0.5,
"max": 1,
"min": 0,
"name": "noise",
"guiName": "Reduce Noise"
},
{
"default": 0.5,
"max": 1,
"min": 0,
"name": "details",
"guiName": "Improve Details"
},
{
"default": 0.5,
"max": 1,
"min": 0,
"name": "halo",
"guiName": "Dehalo"
},
{
"default": 0.5,
"max": 1,
"min": 0,
"name": "blur",
"guiName": "Sharpen"
},
{
"default": 0.5,
"max": 1,
"min": 0,
"name": "compression",
"guiName": "Fix Compression"
}
],
"extraParameters" : [
{
"default": 0,
"max": 0.1,
"min": 0,
"name": "Add Noise",
"guiName": "Add Noise"
}
],
"postflight": 1,
"preflight": 5,
"processors": [
"up",
"aup"
],
"shortName": "rhea",
"version": "1"
}