sunshangquan
commited on
Commit
·
d12e905
1
Parent(s):
5fbced4
- basicsr/metrics/__pycache__/__init__.cpython-38.pyc +0 -0
- basicsr/metrics/__pycache__/metric_util.cpython-38.pyc +0 -0
- basicsr/metrics/__pycache__/niqe.cpython-38.pyc +0 -0
- basicsr/metrics/__pycache__/psnr_ssim.cpython-38.pyc +0 -0
- basicsr/models/__pycache__/__init__.cpython-38.pyc +0 -0
- basicsr/models/__pycache__/base_model.cpython-38.pyc +0 -0
- basicsr/models/__pycache__/image_restoration_model.cpython-38.pyc +0 -0
- basicsr/models/__pycache__/lr_scheduler.cpython-38.pyc +0 -0
- basicsr/models/archs/__pycache__/__init__.cpython-38.pyc +0 -0
- basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc +0 -0
- basicsr/models/image_restoration_model.py +1 -5
- basicsr/models/losses/__pycache__/__init__.cpython-38.pyc +0 -0
- basicsr/models/losses/__pycache__/loss_util.cpython-38.pyc +0 -0
- basicsr/models/losses/__pycache__/losses.cpython-38.pyc +0 -0
- basicsr/utils/__pycache__/__init__.cpython-38.pyc +0 -0
- basicsr/utils/__pycache__/dist_util.cpython-38.pyc +0 -0
- basicsr/utils/__pycache__/file_client.cpython-38.pyc +0 -0
- basicsr/utils/__pycache__/img_util.cpython-38.pyc +0 -0
- basicsr/utils/__pycache__/logger.cpython-38.pyc +0 -0
- basicsr/utils/__pycache__/matlab_functions.cpython-38.pyc +0 -0
- basicsr/utils/__pycache__/misc.cpython-38.pyc +0 -0
- demo.py +1 -1
basicsr/metrics/__pycache__/__init__.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/metrics/__pycache__/__init__.cpython-38.pyc and b/basicsr/metrics/__pycache__/__init__.cpython-38.pyc differ
|
|
|
basicsr/metrics/__pycache__/metric_util.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/metrics/__pycache__/metric_util.cpython-38.pyc and b/basicsr/metrics/__pycache__/metric_util.cpython-38.pyc differ
|
|
|
basicsr/metrics/__pycache__/niqe.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/metrics/__pycache__/niqe.cpython-38.pyc and b/basicsr/metrics/__pycache__/niqe.cpython-38.pyc differ
|
|
|
basicsr/metrics/__pycache__/psnr_ssim.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/metrics/__pycache__/psnr_ssim.cpython-38.pyc and b/basicsr/metrics/__pycache__/psnr_ssim.cpython-38.pyc differ
|
|
|
basicsr/models/__pycache__/__init__.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/__pycache__/__init__.cpython-38.pyc and b/basicsr/models/__pycache__/__init__.cpython-38.pyc differ
|
|
|
basicsr/models/__pycache__/base_model.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/__pycache__/base_model.cpython-38.pyc and b/basicsr/models/__pycache__/base_model.cpython-38.pyc differ
|
|
|
basicsr/models/__pycache__/image_restoration_model.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/__pycache__/image_restoration_model.cpython-38.pyc and b/basicsr/models/__pycache__/image_restoration_model.cpython-38.pyc differ
|
|
|
basicsr/models/__pycache__/lr_scheduler.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/__pycache__/lr_scheduler.cpython-38.pyc and b/basicsr/models/__pycache__/lr_scheduler.cpython-38.pyc differ
|
|
|
basicsr/models/archs/__pycache__/__init__.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/archs/__pycache__/__init__.cpython-38.pyc and b/basicsr/models/archs/__pycache__/__init__.cpython-38.pyc differ
|
|
|
basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc and b/basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc differ
|
|
|
basicsr/models/image_restoration_model.py
CHANGED
|
@@ -50,11 +50,7 @@ class Mixing_Augment:
|
|
| 50 |
target, input_ = self.augments[augment](target, input_)
|
| 51 |
return target, input_
|
| 52 |
|
| 53 |
-
class ImageCleanModel(BaseModel
|
| 54 |
-
PyTorchModelHubMixin,
|
| 55 |
-
library_name="Histoformer",
|
| 56 |
-
repo_url="https://github.com/sunshangquan/Histoformer/",
|
| 57 |
-
docs_url="https://github.com/sunshangquan/Histoformer/",):
|
| 58 |
"""Base Deblur model for single image deblur."""
|
| 59 |
|
| 60 |
def __init__(self, opt):
|
|
|
|
| 50 |
target, input_ = self.augments[augment](target, input_)
|
| 51 |
return target, input_
|
| 52 |
|
| 53 |
+
class ImageCleanModel(BaseModel):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
"""Base Deblur model for single image deblur."""
|
| 55 |
|
| 56 |
def __init__(self, opt):
|
basicsr/models/losses/__pycache__/__init__.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/losses/__pycache__/__init__.cpython-38.pyc and b/basicsr/models/losses/__pycache__/__init__.cpython-38.pyc differ
|
|
|
basicsr/models/losses/__pycache__/loss_util.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/losses/__pycache__/loss_util.cpython-38.pyc and b/basicsr/models/losses/__pycache__/loss_util.cpython-38.pyc differ
|
|
|
basicsr/models/losses/__pycache__/losses.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/models/losses/__pycache__/losses.cpython-38.pyc and b/basicsr/models/losses/__pycache__/losses.cpython-38.pyc differ
|
|
|
basicsr/utils/__pycache__/__init__.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/utils/__pycache__/__init__.cpython-38.pyc and b/basicsr/utils/__pycache__/__init__.cpython-38.pyc differ
|
|
|
basicsr/utils/__pycache__/dist_util.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/utils/__pycache__/dist_util.cpython-38.pyc and b/basicsr/utils/__pycache__/dist_util.cpython-38.pyc differ
|
|
|
basicsr/utils/__pycache__/file_client.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/utils/__pycache__/file_client.cpython-38.pyc and b/basicsr/utils/__pycache__/file_client.cpython-38.pyc differ
|
|
|
basicsr/utils/__pycache__/img_util.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/utils/__pycache__/img_util.cpython-38.pyc and b/basicsr/utils/__pycache__/img_util.cpython-38.pyc differ
|
|
|
basicsr/utils/__pycache__/logger.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/utils/__pycache__/logger.cpython-38.pyc and b/basicsr/utils/__pycache__/logger.cpython-38.pyc differ
|
|
|
basicsr/utils/__pycache__/matlab_functions.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/utils/__pycache__/matlab_functions.cpython-38.pyc and b/basicsr/utils/__pycache__/matlab_functions.cpython-38.pyc differ
|
|
|
basicsr/utils/__pycache__/misc.cpython-38.pyc
CHANGED
|
Binary files a/basicsr/utils/__pycache__/misc.cpython-38.pyc and b/basicsr/utils/__pycache__/misc.cpython-38.pyc differ
|
|
|
demo.py
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
from basicsr.models.archs.histoformer_arch import Histoformer
|
| 2 |
# from huggingface_hub import PyTorchModelHubMixin
|
| 3 |
-
|
| 4 |
# define model
|
| 5 |
model = Histoformer()
|
| 6 |
|
|
|
|
| 1 |
from basicsr.models.archs.histoformer_arch import Histoformer
|
| 2 |
# from huggingface_hub import PyTorchModelHubMixin
|
| 3 |
+
from huggingface_hub import hf_hub_download
|
| 4 |
# define model
|
| 5 |
model = Histoformer()
|
| 6 |
|