我正在编写一个运行长度图像编码器的作业。我的代码对于二进制和8位图像非常有效,但是当我想要编码4位图像时,它无法正确工作。我正在使用Ubuntu 13.10、Python 3.3.4和Pillow。执行以下代码应该打印True和True,但它打印True和False。
为了测试我的代码,我生成了一个随机列表来模拟一张图片。它可以工作,第一个值表示它,但是这个image不起作用。还有4张类似的图片,但它们都不起作用。我错过了什么吗?
为了测试我的代码,我生成了一个随机列表来模拟一张图片。它可以工作,第一个值表示它,但是这个image不起作用。还有4张类似的图片,但它们都不起作用。我错过了什么吗?
from PIL import Image
import random
def _encodeImage4bit(imagePixels, width, height):
encodedImage = bytearray()
count = 0
prev = imagePixels[0]
tempmap = ""
for pixel in imagePixels:
if count >= 15:
encodedImage.append(15)
encodedImage.append(pixel)
tempmap += "1"
tempmap += "0"
count = 0
if pixel == prev:
count += 1
else:
if count > 1:
encodedImage.append(count)
tempmap += "1"
encodedImage.append(prev)
tempmap += "0"
count = 1
prev = pixel
if count > 1:
encodedImage.append(count)
tempmap += "1"
encodedImage.append(prev)
tempmap += "0"
encodedImage.extend([0] * _remaining(len(encodedImage)))
tempmap += "1"*_remaining(len(tempmap))
encodedImage = _set4bitMap(tempmap, encodedImage)
return encodedImage
def _set4bitMap(imgMap, encodedImage):
newImgMap = _divideByRow(imgMap, 8)
tempImg = [_merge4bitTo8bit(encodedImage[i], encodedImage[i + 1]) for i in range(0, len(encodedImage), 2)]
tempImg = _divideByRow(list(tempImg), 4)
return bytearray(_flattenListOfList(_mergeMap(tempImg, newImgMap)))
def _decodeImage4bit(encodedImage, width, height):
decodedImage = []
imgMap, encImg = _get4bitMap(encodedImage)
for index, i in enumerate(imgMap):
if i == '1' and encImg[index] == 0:
break
if i == '1':
decodedImage.extend([encImg[index + 1]] * encImg[index])
elif imgMap[index - 1] != '1' or index == 0:
decodedImage.append(encImg[index])
return decodedImage
def _get4bitMap(encodedImage):
imgMap = ""
newEncodedImage = list(encodedImage)
I = range(0, len(newEncodedImage), 5)
for i in I:
imgMap += '{0:08b}'.format(newEncodedImage[i])
for i in sorted(list(I), reverse = True):
del newEncodedImage[i]
newEncodedImage = _flattenListOfList([_split8bitTo4bit(i) for i in newEncodedImage])
return (imgMap, newEncodedImage)
def _split8bitTo4bit(eightbit):
leftmask = 240
rightmask = 15
left = (eightbit & leftmask) >> 4
right = eightbit & rightmask
return (left, right)
def _merge4bitTo8bit(left, right):
return (left << 4) | right
_remaining = lambda x, y = 8: 0 if x % y == 0 else y - (x % y)
_mergeMap = lambda z, x:[[int(x[index], 2)] + i for index, i in enumerate(z)]
_flattenListOfList = lambda flat:[item for sublist in flat for item in sublist]
_divideByRow = lambda flat, size: [flat[i:i + size] for i in range(0, len(flat), size)]
if __name__ == "__main__":
img = [15] * 100
img.extend([random.randrange(0, 16) for n in range(300)])
encImg = _encodeImage4bit(img, 20, 20)
decImg = _decodeImage4bit(encImg, 20, 20)
print(str(decImg == img))
imgpath = "../../images/4bit/baboon_4bit.bmp"
img2 = Image.open(imgpath)
encImg2 = _encodeImage4bit(list(img2.getdata(0)), img2.size[0], img2.size[1])
decImg2 = _decodeImage4bit(encImg2, img2.size[0], img2.size[1])
print(str(decImg2 == list(img2.getdata(0))))
来自数据压缩完全参考第4版第26页的算法
同样,每个字节都有一个比特位来指示它是否包含灰度值或计数。然而,这一次,这些额外的比特位被累积成8个一组,并且每组在对应的8个字节之前(或之后)写入输出流中。
我修改了它以支持4位图像。
例子 原始图像:12、12、12、12、12、12、12、12、12、14、3、7、10、10、10、10、5、5、5、5、5、5、1、……
第一步:查找重复项 9、12、14、3、7、4、10、6、5、1、……
第二步:生成地图以确定哪个元素是像素值(0),哪个元素是重复次数(1) 1 0 0 0 0 1 0 1 0 ……
第三步: 用零填充值,使长度成为8的倍数。 用1填充地图,使长度成为8的倍数。
第四步: 将地图分成八块,并将每个块转换为整数。 133,……
第五步: 每两个4位组合成一个8位。将左值向左移4次并与右数进行OR运算。 156、227、116、166、81、……
第六步: 将地图与值合并。现在,地图中的每个整数数字表示值中的4个值。 133、156、227、116、166、……
解码是这个操作的反向。