ASCEE/pyqtgraph
5
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix ImageItem rgb histogram calculation

Browse Source
This commit is contained in:
Luke Campagnola 2017-10-03 08:16:36 -07:00
parent fcf4503671
commit c3e52f15b0
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
pyqtgraph/graphicsItems/ImageItem.py
View File

@ -489,14 +489,17 @@ class ImageItem(GraphicsObject):
bins = [mn, mx]
kwds['bins'] = bins
stepData = stepData[np.isfinite(stepData)]
if perChannel:
hist = []
for i in range(stepData.shape[-1]):
h = np.histogram(stepData[..., i], **kwds)
stepChan = stepData[..., i]
stepChan = stepChan[np.isfinite(stepChan)]
h = np.histogram(stepChan, **kwds)
hist.append((h[1][:-1], h[0]))
return hist
else:
stepData = stepData[np.isfinite(stepData)]
hist = np.histogram(stepData, **kwds)
return hist[1][:-1], hist[0]