c

Browse files

Files changed (3) hide show

basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc +0 -0
basicsr/models/archs/histoformer_arch.py +1 -1
demo.py +1 -1

basicsr/models/archs/__pycache__/histoformer_arch.cpython-38.pyc CHANGED Viewed

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/archs/histoformer_arch.py CHANGED Viewed

@@ -267,7 +267,7 @@ class Upsample(nn.Module):
         return self.body(x)
 ##########################################################################
-class Histoformer(nn.Module, PyTorchModelHubMixin, repo_url="https://github.com/sunshangquan/Histoformer",):
     def __init__(self,
         inp_channels=3,
         out_channels=3,

         return self.body(x)
 ##########################################################################
+class Histoformer(nn.Module, PyTorchModelHubMixin, ):
     def __init__(self,
         inp_channels=3,
         out_channels=3,

demo.py CHANGED Viewed

@@ -14,4 +14,4 @@ model.load_state_dict(torch.load(filepath, map_location='cpu')['params'], )
 model.push_to_hub("sunsean/Histoformer-real")
 # reload
-model = ImageCleanModel.from_pretrained("sunsean/Histoformer-real")

 model.push_to_hub("sunsean/Histoformer-real")
 # reload
+model = Histoformer.from_pretrained("sunsean/Histoformer-real")