问题
I am using Python 3.5 and OpenCV 3 to analyze pictures of cells in biology. My pictures look like this:
I want to be able to calculate a ratio of the area of the cell's nucleus to the area of the cell as a whole.
In my slides, the nucleus is dark purple and other regions of the cell are light blue. There are also tan coloured red blood cells which I want to ignore entirely. For clarity, here's a labeled image:
How can I use image segmentation to identify and measure my regions of interest?
I have tried follow this guide, but it returns a completely black image.
回答1:
First, some preliminary code that we'll use below:
import numpy as np
import cv2
from matplotlib import pyplot as plt
from skimage.morphology import extrema
from skimage.morphology import watershed as skwater
def ShowImage(title,img,ctype):
if ctype=='bgr':
b,g,r = cv2.split(img) # get b,g,r
rgb_img = cv2.merge([r,g,b]) # switch it to rgb
plt.imshow(rgb_img)
elif ctype=='hsv':
rgb = cv2.cvtColor(img,cv2.COLOR_HSV2RGB)
plt.imshow(rgb)
elif ctype=='gray':
plt.imshow(img,cmap='gray')
elif ctype=='rgb':
plt.imshow(img)
else:
raise Exception("Unknown colour type")
plt.title(title)
plt.show()
For reference, here's your original image:
#Read in image
img = cv2.imread('cells.jpg')
ShowImage('Original',img,'bgr')
The article you've linked to suggests using Otsu's method for colour segmentation. The method assumes that the intensity of the pixels of the image can be plotted into a bimodal histogram, and finds an optimal separator for that histogram. I apply the method below.
#Convert to a single, grayscale channel
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
#Threshold the image to binary using Otsu's method
ret, thresh = cv2.threshold(gray,0,255,cv2.THRESH_BINARY_INV+cv2.THRESH_OTSU)
ShowImage('Grayscale',gray,'gray')
ShowImage('Applying Otsu',thresh,'gray')
The binary form of the image isn't that good! Looking at the grayscale image, you can see why: the Otsu transformation produces three classes of pixels: the dark background pixels, the donut cells and cell interiors, and the nuclei. The histogram below demonstrates this:
#Make a histogram of the intensities in the grayscale image
plt.hist(gray.ravel(),256)
plt.show()
Thus, you've broken the assumptions of the algorithm you are using, so it's not unexpected that you get bad results. By throwing away colour information, we've lost the ability to differentiate donuts from cell interiors.
One way of dealing with this is to perform segmentation based on colour thresholding. To do this, you choose a color space to work in. This guide has an excellent pictorial description of the different spaces.
Let's choose HSV. This has the advantage that a single channel, H, describes the image's colour. Once we've converted our image into this space, we can find the bounds of our colours of interest. For instance, to find the cell's nuclei we could do as follows:
cell_hsvmin = (110,40,145) #Lower end of the HSV range defining the nuclei
cell_hsvmax = (150,190,255) #Upper end of the HSV range defining the nuclei
#Transform image to HSV color space
hsv = cv2.cvtColor(img,cv2.COLOR_BGR2HSV)
#Threshold based on HSV values
color_thresh = cv2.inRange(hsv, cell_hsvmin, cell_hsvmax)
ShowImage('Color Threshold',color_thresh,'gray')
masked = cv2.bitwise_and(img,img, mask=color_thresh)
ShowImage('Color Threshold Maksed',masked,'bgr')
This looks much better! Though, notice that some portions of the cell's interiors are being labeled as nucleii, even though they should not. One could also argue that it's not very automatic: you still have to carefully handpick your colours. Operating in HSV space eliminates a lot of the guesswork, but maybe we could use the fact that there are four distinct colours to obviate the need for ranges! To do so, we pass our HSV pixels through a k-means clustering algorithm.
#Convert pixel space to an array of triplets. These are vectors in 3-space.
Z = hsv.reshape((-1,3))
#Convert to floating point
Z = np.float32(Z)
#Define the K-means criteria, these are not too important
criteria = (cv2.TERM_CRITERIA_EPS + cv2.TERM_CRITERIA_MAX_ITER, 10, 1.0)
#Define the number of clusters to find
K = 4
#Perform the k-means transformation. What we get back are:
#*Centers: The coordinates at the center of each 3-space cluster
#*Labels: Numeric labels for each cluster
#*Ret: A return code indicating whether the algorithm converged, &c.
ret,label,center = cv2.kmeans(Z,K,None,criteria,10,cv2.KMEANS_RANDOM_CENTERS)
#Produce an image using only the center colours of the clusters
center = np.uint8(center)
khsv = center[label.flatten()]
khsv = khsv.reshape((img.shape))
ShowImage('K-means',khsv,'hsv')
#Reshape labels for masking
label = label.reshape(img.shape[0:2])
ShowImage('K-means Labels',label,'gray')
Notice that this has done an excellent job of separting the colours without the need for manual specification! (Other than specifying the number of clusters.)
Now, we need to figure out which labels correspond to which parts of the cell.
To do so, we find the colour of two pixels: one which is clearly a nucleus pixel and one which is clearly a cell pixel. We then figure out which cluster center is closest to each of these pixels.
#(Distance,Label) pairs
nucleus_colour = np.array([139, 106, 192])
cell_colour = np.array([130, 41, 207])
nuclei_label = (np.inf,-1)
cell_label = (np.inf,-1)
for l,c in enumerate(center):
print(l,c)
dist_nuc = np.sum(np.square(c-nucleus_colour)) #Euclidean distance between colours
if dist_nuc<nuclei_label[0]:
nuclei_label=(dist_nuc,l)
dist_cell = np.sum(np.square(c-cell_colour)) #Euclidean distance between colours
if dist_cell<cell_label[0]:
cell_label=(dist_cell,l)
nuclei_label = nuclei_label[1]
cell_label = cell_label[1]
print("Nuclei label={0}, cell label={1}".format(nuclei_label,cell_label))
Now, let's build the binary classifier we need to identify the entirety of the cells for the watershed algorithm:
#Multiply by 1 to keep image in an integer format suitable for OpenCV
thresh = cv2.bitwise_or(1*(label==nuclei_label),1*(label==cell_label))
thresh = np.uint8(thresh)
ShowImage('Binary',thresh,'gray')
We can now eliminate single-pixel noise:
#Remove noise by eliminating single-pixel patches
kernel = np.ones((3,3),np.uint8)
opening = cv2.morphologyEx(thresh,cv2.MORPH_OPEN, kernel, iterations = 2)
ShowImage('Opening',opening,'gray')
We now need to identify the peaks of the watershed and give them separate labels. The goal of this is to generate a set of pixels such that each of the nuclei+cells has a pixel within it and no two nuclei have their identifier pixels touching.
To achieve this, we could perform a distance transformation and then filter out distances that are two far from the center of the nuclei+cells.
However, we have to be careful since long, narrow cells with high thresholds might disappear entirely. In the image below, we've separated the two cells that were touching in the bottom right, but entirely eliminated the long, narrow cell that was in the upper right.
#Identify areas which are surely foreground
fraction_foreground = 0.75
dist = cv2.distanceTransform(opening,cv2.DIST_L2,5)
ret, sure_fg = cv2.threshold(dist,fraction_foreground*dist.max(),255,0)
ShowImage('Distance',dist_transform,'gray')
ShowImage('Surely Foreground',sure_fg,'gray')
Decreasing the threshold brings the long, narrow cell back, but leaves the cells in the bottom right connected.
We can fix this by using an adaptive method which identifies the peaks in each local area. This eliminates the need to set a single, global constant for our threshold. To do so, we use the h_axima function, which returns all local maxima that are greater than a specified cut-off value. This contrasts with the distance function, which returned all pixels greater than a given value.
#Identify areas which are surely foreground
h_fraction = 0.1
dist = cv2.distanceTransform(opening,cv2.DIST_L2,5)
maxima = extrema.h_maxima(dist, h_fraction*dist.max())
print("Peaks found: {0}".format(np.sum(maxima)))
#Dilate the maxima so we can see them
maxima = cv2.dilate(maxima, kernel, iterations=2)
ShowImage('Distance',dist_transform,'gray')
ShowImage('Surely Foreground',maxima,'gray')
Now we identify unknown regions, the regions which will be labeled by the watershed algorithm, by subtracting off the maxima:
# Finding unknown region
unknown = cv2.subtract(opening,maxima)
ShowImage('Unknown',unknown,'gray')
Next, we give each of the maxima unique labels and then mark the unknown regions before finally performing the watershed transform:
# Marker labelling
ret, markers = cv2.connectedComponents(maxima)
ShowImage('Connected Components',markers,'rgb')
# Add one to all labels so that sure background is not 0, but 1
markers = markers+1
# Now, mark the region of unknown with zero
markers[unknown==np.max(unknown)] = 0
ShowImage('markers',markers,'rgb')
dist = cv2.distanceTransform(opening,cv2.DIST_L2,5)
markers = skwater(-dist,markers,watershed_line=True)
ShowImage('Watershed',markers,'rgb')
imgout = img.copy()
imgout[markers == 0] = [0,0,255] #Label the watershed_line
ShowImage('img',imgout,'bgr')
This gives us a set of labeled regions representing the cells. Next, we iterate through these regions, use them as masks on our labeled data, and calculate the fractions:
for l in np.unique(markers):
if l==0: #Watershed line
continue
if l==1: #Background
continue
#For displaying individual cells
#temp=khsv.copy()
#temp[markers!=l]=0
#ShowImage('out',temp,'hsv')
temp = label.copy()
temp[markers!=l]=-1
nucleus_area = np.sum(temp==nuclei_label)
cell_area = np.sum(temp==cell_label)
print("Nucleus fraction for cell {0} is {1}".format(l,nucleus_area/(cell_area+nucleus_area)))
This gives:
Nucleus fraction for cell 2 is 0.9002795899347623
Nucleus fraction for cell 3 is 0.7953321364452424
Nucleus fraction for cell 4 is 0.7525925925925926
Nucleus fraction for cell 5 is 0.8151515151515152
Nucleus fraction for cell 6 is 0.6808656818962556
Nucleus fraction for cell 7 is 0.8276481149012568
Nucleus fraction for cell 8 is 0.878500237304224
Nucleus fraction for cell 9 is 0.8342518016108521
Nucleus fraction for cell 10 is 0.9742324561403509
Nucleus fraction for cell 11 is 0.8728733459357277
Nucleus fraction for cell 12 is 0.7968570333461096
Nucleus fraction for cell 13 is 0.8226831716293075
Nucleus fraction for cell 14 is 0.7491039426523297
Nucleus fraction for cell 15 is 0.839096357768557
Nucleus fraction for cell 16 is 0.7589670014347202
Nucleus fraction for cell 17 is 0.8559168925022583
Nucleus fraction for cell 18 is 0.7534142640364189
Nucleus fraction for cell 19 is 0.8036734693877551
Nucleus fraction for cell 20 is 0.7566037735849057
(Note that if you're using this for academic purposes, academic integrity requires proper attribution. Please contact me for details.)
回答2:
# light purple color segmentation (to get cells)
cell_hsvmin = (110,40,145)
cell_hsvmax = (150,190,255)
hsv = cv2.cvtColor(frame, cv2.COLOR_BGR2HSV)
color_thresh = cv2.inRange(hsv, cell_hsvmin, cell_hsvmax)
# masked = cv2.bitwise_and(frame,frame, mask=color_thresh)
# cv2.imshow('masked0', masked)
ksize = 5
open_thresh = cv2.morphologyEx(color_thresh, cv2.MORPH_OPEN, np.ones((ksize,ksize),'uint8'), iterations=1)
masked = cv2.bitwise_and(frame,frame, mask=open_thresh)
cv2.imshow('masked', masked)
# dark purple color segmentation (to get nucleus)
nucleus_hsvmin = (125,65,160)
nucleus_hsvmax = (150,190,255)
nucleus_color_thresh = cv2.inRange(hsv, nucleus_hsvmin, nucleus_hsvmax)
ksize = 3
nucleus_open_thresh = cv2.morphologyEx(nucleus_color_thresh, cv2.MORPH_OPEN, np.ones((ksize,ksize),'uint8'), iterations=1)
nucleus_masked = cv2.bitwise_and(masked,masked, mask=nucleus_open_thresh)
cv2.imshow('nucleus_masked', nucleus_masked)
"""
HULL APPROXIMATES THE CELLS TO A CIRCLE TO FILL IN GAPS CREATED BY THRESHOLDING AND CLOSING.
FOR NON-CIRCULAR CELLS LIKE IN YOUR SECOND IMAGE, THIS MIGHT CAUSE BAD AREA CALCULATIONS
"""
# doHULL = False
doHULL = True
cells = []
cells_ratio = []
minArea = frame.shape[0]*frame.shape[1]* 0.01
_, contours, _ = cv2.findContours(open_thresh,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_SIMPLE)
for cnt in contours:
area = cv2.contourArea(cnt)
if area >= minArea:
cells.append(cnt)
nucleus_mask = np.zeros(frame.shape[:2], 'uint8')
if doHULL:
hull = cv2.convexHull(cnt)
cv2.drawContours(nucleus_mask, [hull], -1, 255, -1);
else:
cv2.drawContours(nucleus_mask, [cnt], -1, 255, -1);
nucleus_mask = cv2.bitwise_and(nucleus_open_thresh, nucleus_mask)
nucleus_area = np.count_nonzero(nucleus_mask)
ratio = nucleus_area / area
cells_ratio.append(ratio)
# nucleus_img = cv2.bitwise_and(frame, frame, mask=nucleus_mask)
# cv2.imshow('nucleus_img', nucleus_img)
# cv2.waitKey(0)
doDRAWCELLS = False
# doDRAWCELLS = True
if doDRAWCELLS:
for cell_cnt in cells:
cells_mask = np.zeros(frame.shape[:2], 'uint8')
if doHULL:
hull = cv2.convexHull(cell_cnt)
cv2.drawContours(cells_mask, [hull], -1, 255, -1);
else:
cv2.drawContours(cells_mask, [cell_cnt], -1, 255, -1);
cells_img = cv2.bitwise_and(frame, frame, mask=cells_mask)
cv2.imshow('cells_img', cells_img)
cv2.waitKey(0)
this will only work for cells that are not connected. you can use this as a base to work with the watershed algorithm. Also, the color segmentation parameters have been tuned according to the 2 images you posted. other slides might deviate from the color range so you might have to adjust them. if adjusting them doesnt get you a good compromise, you might have to look into otsu binarization or adaptive thresholding to segment the colors.
Another option is to look at cv2.MORPH_GRADIENT which works like an edge detector. or
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
kernel = np.array([[1,1,1],[1,-8,1],[1,1,1]],dtype='float32')
laplace = cv2.filter2D(cv2.GaussianBlur(gray,(blur_ksize,blur_ksize),0), -1, kernel)
cv2.imshow('laplace', laplace)
and use the edges to segment the cells?
来源：https://stackoverflow.com/questions/51621684/how-to-calculate-nucleus-amount-of-cell