Spaces:
Runtime error
Runtime error
Commit
·
72bc1c2
1
Parent(s):
471186c
Auto deploy
Browse files- stCompressService.py +3 -5
stCompressService.py
CHANGED
@@ -46,8 +46,7 @@ def compressImage(image: torch.Tensor, model: BaseCompressor, crop: bool) -> Fil
|
|
46 |
# [c, h, w]
|
47 |
image = (image - 0.5) * 2
|
48 |
|
49 |
-
|
50 |
-
codes, binaries, headers = model.compress(image[None, ...], cdfs)
|
51 |
|
52 |
return File(headers[0], binaries[0])
|
53 |
|
@@ -56,9 +55,8 @@ def compressImage(image: torch.Tensor, model: BaseCompressor, crop: bool) -> Fil
|
|
56 |
def decompressImage(sourceFile: File, model: BaseCompressor) -> torch.ByteTensor:
|
57 |
binaries = sourceFile.Content
|
58 |
|
59 |
-
|
60 |
-
|
61 |
-
restored = model.decompress([binaries], cdfs, [sourceFile.FileHeader])
|
62 |
|
63 |
# [c, h, w]
|
64 |
return DeTransform()(restored[0])
|
|
|
46 |
# [c, h, w]
|
47 |
image = (image - 0.5) * 2
|
48 |
|
49 |
+
codes, binaries, headers = model.compress(image[None, ...])
|
|
|
50 |
|
51 |
return File(headers[0], binaries[0])
|
52 |
|
|
|
55 |
def decompressImage(sourceFile: File, model: BaseCompressor) -> torch.ByteTensor:
|
56 |
binaries = sourceFile.Content
|
57 |
|
58 |
+
# [1, c, h, w]
|
59 |
+
restored = model.decompress([binaries], [sourceFile.FileHeader])
|
|
|
60 |
|
61 |
# [c, h, w]
|
62 |
return DeTransform()(restored[0])
|