xiaosu-zhu commited on
Commit
72bc1c2
·
1 Parent(s): 471186c

Auto deploy

Browse files
Files changed (1) hide show
  1. 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
- with model.readyForCoding() as cdfs:
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
- with model.readyForCoding() as cdfs:
60
- # [1, c, h, w]
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])