def metrics(self):
progress = QProgressDialog(self.tr('Computing metrics...'),
self.tr('Cancel'), 1,
self.table_widget.rowCount(), self)
progress.canceled.connect(self.cancel)
progress.setWindowModality(Qt.WindowModal)
img1 = cv.cvtColor(self.evidence, cv.COLOR_BGR2GRAY)
img2 = cv.cvtColor(self.reference, cv.COLOR_BGR2GRAY)
x = img1.astype(np.float64)
y = img2.astype(np.float64)
rmse = self.rmse(x, y)
progress.setValue(1)
if self.stopped:
return
sam = sewar.sam(img1, img2)
progress.setValue(2)
if self.stopped:
return
ergas = sewar.ergas(img1, img2)
progress.setValue(3)
if self.stopped:
return
mb = self.mb(x, y)
progress.setValue(4)
if self.stopped:
return
pfe = self.pfe(x, y)
progress.setValue(5)
if self.stopped:
return
psnr = self.psnr(x, y)
progress.setValue(6)
if self.stopped:
return
try:
psnrb = sewar.psnrb(img1, img2)
except NameError:
# FIXME: C'\`e un bug in psnrb (https://github.com/andrewekhalel/sewar/issues/17)
psnrb = 0
progress.setValue(7)
if self.stopped:
return
ssim, self.ssim_map = self.ssim(x, y)
progress.setValue(8)
if self.stopped:
return
mssim = sewar.msssim(img1, img2).real
progress.setValue(9)
if self.stopped:
return
rase = sewar.rase(img1, img2)
progress.setValue(10)
if self.stopped:
return
scc = sewar.scc(img1, img2)
progress.setValue(11)
if self.stopped:
return
uqi = sewar.uqi(img1, img2)
progress.setValue(12)
if self.stopped:
return
vifp = sewar.vifp(img1, img2)
progress.setValue(13)
if self.stopped:
return
ssimul = self.ssimul(img1, img2)
progress.setValue(14)
if self.stopped:
return
butter, self.butter_map = self.butter(img1, img2)
progress.setValue(15)
if self.stopped:
return
sizes = [256, 256, 256]
ranges = [0, 256] * 3
channels = [0, 1, 2]
hist1 = cv.calcHist([self.evidence], channels, None, sizes, ranges)
hist2 = cv.calcHist([self.reference], channels, None, sizes, ranges)
correlation = cv.compareHist(hist1, hist2, cv.HISTCMP_CORREL)
progress.setValue(16)
if self.stopped:
return
chi_square = cv.compareHist(hist1, hist2, cv.HISTCMP_CHISQR)
progress.setValue(17)
if self.stopped:
return
chi_square2 = cv.compareHist(hist1, hist2, cv.HISTCMP_CHISQR_ALT)
progress.setValue(18)
if self.stopped:
return
intersection = cv.compareHist(hist1, hist2, cv.HISTCMP_INTERSECT)
progress.setValue(19)
if self.stopped:
return
hellinger = cv.compareHist(hist1, hist2, cv.HISTCMP_HELLINGER)
progress.setValue(20)
if self.stopped:
return
divergence = cv.compareHist(hist1, hist2, cv.HISTCMP_KL_DIV)
progress.setValue(21)
self.table_widget.setItem(0, 1,
QTableWidgetItem('{:.2f}'.format(rmse)))
self.table_widget.setItem(1, 1, QTableWidgetItem('{:.4f}'.format(sam)))
self.table_widget.setItem(2, 1,
QTableWidgetItem('{:.2f}'.format(ergas)))
self.table_widget.setItem(3, 1, QTableWidgetItem('{:.4f}'.format(mb)))
self.table_widget.setItem(4, 1, QTableWidgetItem('{:.2f}'.format(pfe)))
if psnr > 0:
self.table_widget.setItem(
5, 1, QTableWidgetItem('{:.2f} dB'.format(psnr)))
else:
self.table_widget.setItem(
5, 1, QTableWidgetItem('+' + u'\u221e' + ' dB'))
self.table_widget.setItem(6, 1,
QTableWidgetItem('{:.2f}'.format(psnrb)))
self.table_widget.setItem(7, 1,
QTableWidgetItem('{:.4f}'.format(ssim)))
self.table_widget.setItem(8, 1,
QTableWidgetItem('{:.4f}'.format(mssim)))
self.table_widget.setItem(9, 1,
QTableWidgetItem('{:.2f}'.format(rase)))
self.table_widget.setItem(10, 1,
QTableWidgetItem('{:.4f}'.format(scc)))
self.table_widget.setItem(11, 1,
QTableWidgetItem('{:.4f}'.format(uqi)))
self.table_widget.setItem(12, 1,
QTableWidgetItem('{:.4f}'.format(vifp)))
self.table_widget.setItem(13, 1,
QTableWidgetItem('{:.4f}'.format(ssimul)))
self.table_widget.setItem(14, 1,
QTableWidgetItem('{:.2f}'.format(butter)))
self.table_widget.setItem(
15, 1, QTableWidgetItem('{:.2f}'.format(correlation)))
self.table_widget.setItem(
16, 1, QTableWidgetItem('{:.2f}'.format(chi_square)))
self.table_widget.setItem(
17, 1, QTableWidgetItem('{:.2f}'.format(chi_square2)))
self.table_widget.setItem(
18, 1, QTableWidgetItem('{:.2f}'.format(intersection)))
self.table_widget.setItem(19, 1,
QTableWidgetItem('{:.2f}'.format(hellinger)))
self.table_widget.setItem(
20, 1, QTableWidgetItem('{:.2f}'.format(divergence)))
self.table_widget.resizeColumnsToContents()
self.table_widget.setEnabled(True)
self.metric_button.setEnabled(False)
self.ssim_radio.setEnabled(True)
self.butter_radio.setEnabled(True)
def test_color(self):
sam = sewar.sam(self.read('clr'), self.read('clr'))
self.assertTrue(sam < self.eps)
def test_gray(self):
sam = sewar.sam(self.read('gry'), self.read('gry'))
self.assertTrue(sam < self.eps)
def metrics(self):
progress = QProgressDialog(self.tr('Computing metrics...'),
self.tr('Cancel'), 1,
self.table_widget.rowCount(), self)
progress.canceled.connect(self.cancel)
progress.setWindowModality(Qt.WindowModal)
img1 = cv.cvtColor(self.evidence, cv.COLOR_BGR2GRAY)
img2 = cv.cvtColor(self.reference, cv.COLOR_BGR2GRAY)
x = img1.astype(np.float64)
y = img2.astype(np.float64)
rmse = self.rmse(x, y)
progress.setValue(1)
if self.stopped:
return
sam = sewar.sam(img1, img2)
progress.setValue(2)
if self.stopped:
return
ergas = sewar.ergas(img1, img2)
progress.setValue(3)
if self.stopped:
return
mb = self.mb(x, y)
progress.setValue(4)
if self.stopped:
return
pfe = self.pfe(x, y)
progress.setValue(5)
if self.stopped:
return
psnr = self.psnr(x, y)
progress.setValue(6)
if self.stopped:
return
psnrb = sewar.psnrb(img1, img2)
progress.setValue(7)
if self.stopped:
return
ssim, self.ssim_map = self.ssim(x, y)
progress.setValue(8)
if self.stopped:
return
mssim = sewar.msssim(img1, img2).real
progress.setValue(9)
if self.stopped:
return
rase = sewar.rase(img1, img2)
progress.setValue(10)
if self.stopped:
return
scc = sewar.scc(img1, img2)
progress.setValue(11)
if self.stopped:
return
uqi = sewar.uqi(img1, img2)
progress.setValue(12)
if self.stopped:
return
vifp = sewar.vifp(img1, img2)
progress.setValue(13)
if self.stopped:
return
ssimul = self.ssimul(img1, img2)
progress.setValue(14)
if self.stopped:
return
butter, self.butter_map = self.butter(img1, img2)
progress.setValue(15)
self.table_widget.setItem(0, 1,
QTableWidgetItem('{:.4f}'.format(rmse)))
self.table_widget.setItem(1, 1, QTableWidgetItem('{:.4f}'.format(sam)))
self.table_widget.setItem(2, 1,
QTableWidgetItem('{:.4f}'.format(ergas)))
self.table_widget.setItem(3, 1, QTableWidgetItem('{:.4f}'.format(mb)))
self.table_widget.setItem(4, 1, QTableWidgetItem('{:.4f}'.format(pfe)))
if psnr > 0:
self.table_widget.setItem(
5, 1, QTableWidgetItem('{:.4f} dB'.format(psnr)))
else:
self.table_widget.setItem(
5, 1, QTableWidgetItem('+' + u'\u221e' + ' dB'))
self.table_widget.setItem(6, 1,
QTableWidgetItem('{:.4f}'.format(psnrb)))
self.table_widget.setItem(7, 1,
QTableWidgetItem('{:.4f}'.format(ssim)))
self.table_widget.setItem(8, 1,
QTableWidgetItem('{:.4f}'.format(mssim)))
self.table_widget.setItem(9, 1,
QTableWidgetItem('{:.4f}'.format(rase)))
self.table_widget.setItem(10, 1,
QTableWidgetItem('{:.4f}'.format(scc)))
self.table_widget.setItem(11, 1,
QTableWidgetItem('{:.4f}'.format(uqi)))
self.table_widget.setItem(12, 1,
QTableWidgetItem('{:.4f}'.format(vifp)))
self.table_widget.setItem(13, 1,
QTableWidgetItem('{:.4f}'.format(ssimul)))
self.table_widget.setItem(14, 1,
QTableWidgetItem('{:.4f}'.format(butter)))
self.table_widget.resizeColumnsToContents()
self.table_widget.setEnabled(True)
self.metric_button.setEnabled(False)
self.ssim_radio.setEnabled(True)
self.butter_radio.setEnabled(True)