Spaces:
Build error
Build error
TalhaUsuf
commited on
Commit
•
d7939cf
1
Parent(s):
ad250d1
added the llflow model
Browse files
models/llflow/LOL_smallNet.yml
CHANGED
|
@@ -58,7 +58,7 @@ dataroot_unpaired: models/llflow/dataset_samples/our485/low
|
|
| 58 |
# dataroot_unpaired: /home/data/Dataset/LOL_test/Fusion
|
| 59 |
dataroot_GT: D:\Dataset\LOL-v2\LOL-v2\IntegratedTest\Test\high
|
| 60 |
dataroot_LR: D:\Dataset\LOL-v2\LOL-v2\IntegratedTest\Test\low
|
| 61 |
-
model_path: models
|
| 62 |
heat: 0 # This is the standard deviation of the latent vectors
|
| 63 |
|
| 64 |
#### network structures
|
|
|
|
| 58 |
# dataroot_unpaired: /home/data/Dataset/LOL_test/Fusion
|
| 59 |
dataroot_GT: D:\Dataset\LOL-v2\LOL-v2\IntegratedTest\Test\high
|
| 60 |
dataroot_LR: D:\Dataset\LOL-v2\LOL-v2\IntegratedTest\Test\low
|
| 61 |
+
model_path: E:\startup\image_enhancement\models\llflow\LOL_smallNet.pth
|
| 62 |
heat: 0 # This is the standard deviation of the latent vectors
|
| 63 |
|
| 64 |
#### network structures
|
models/llflow/models/modules/ConditionEncoder.py
CHANGED
|
@@ -7,7 +7,7 @@ import torch.nn as nn
|
|
| 7 |
import torch.nn.functional as F
|
| 8 |
import models.modules.module_util as mutil
|
| 9 |
# from utils.util import opt_get
|
| 10 |
-
from models.modules.flow import Conv2dZeros
|
| 11 |
|
| 12 |
|
| 13 |
def opt_get(opt, keys, default=None):
|
|
|
|
| 7 |
import torch.nn.functional as F
|
| 8 |
import models.modules.module_util as mutil
|
| 9 |
# from utils.util import opt_get
|
| 10 |
+
from models.llflow.models.modules.flow import Conv2dZeros
|
| 11 |
|
| 12 |
|
| 13 |
def opt_get(opt, keys, default=None):
|