from scipy.misc import imread
from matplotlib import pyplot
import cv2
from cv2 import cv
from SRM import SRM ## Module for Statistical Regional Segmentation
im = imread("lena.png")
im2 = cv2.imread("lena.png")
print type(im), type(im2), im.shape, im2.shape
## Prints <type 'numpy.ndarray'> <type 'numpy.ndarray'> (120, 120, 3) (120, 120, 3)
srm = SRM(im, 256)
segmented = srm.run()
srm2 = SRM(im2, 256)
segmented2 = srm2.run()
pic = segmented/256
pic2 = segmented2/256
pyplot.imshow(pic)
pyplot.imsave("onePic.jpg", pic)
pic = pic.astype('uint8')
cv2.imwrite("onePic2.jpg", pic2)
pyplot.show()
onePic.jpg
は正しいセグメント化された画像を提供しますonePic2.jpg
が、完全な黒の画像を提供します。uint8
データ型を使用に変換しpic = pic.astype('uint8')
ても役に立ちませんでした。私はまだ黒いイメージを与えます!
onePic.jpg を使用pyplot.imsave()
:
onePic2.jpg を使用cv2.imwrite()
:
助けてください!