#?PyQt5相關(guān)模塊
from?PyQt5.QtWidgets?import?*
from?PyQt5.QtCore?import?*
from?PyQt5.QtGui?import?*

#?動漫圖片制作的業(yè)務(wù)模塊
import?cv2
import?sys
import?os

#?日志模塊
from?loguru?import?logger

class?CartoonUI(QWidget):
????def?__init__(self):
????????'''
????????初始化UI界面應(yīng)用
????????'''
????????super(CartoonUI,?self).__init__()
????????self.init_ui()

????def?init_ui(self):
????????'''
????????自定義實現(xiàn)的ui應(yīng)用函數(shù)
????????:return:
????????'''
????????self.setWindowTitle('動漫圖片轉(zhuǎn)換器??公眾號：Python 集中營')
????????self.setWindowIcon(QIcon('ico.png'))
????????self.setFixedWidth(500)

????????hbox?=?QHBoxLayout()
????????self.input_image_path?=?QLineEdit()
????????self.input_image_path.setPlaceholderText('源圖片路徑')
????????self.input_image_path.setReadOnly(True)

????????self.input_image_btn?=?QPushButton()
????????self.input_image_btn.setText('導(dǎo)入源圖片')
????????self.input_image_btn.clicked.connect(self.input_image_btn_click)

????????self.generate_btn?=?QPushButton()
????????self.generate_btn.setText('一鍵生成動漫圖片')
????????self.generate_btn.clicked.connect(self.generate_btn_click)

????????hbox.addWidget(self.input_image_path)
????????hbox.addWidget(self.input_image_btn)
????????hbox.addWidget(self.generate_btn)

????????self.thread_?=?WorkThread(self)
????????self.thread_.finished.connect(self.finished)

????????self.setLayout(hbox)

????def?input_image_btn_click(self):
????????'''
????????input_image_btn按鈕綁定的槽函數(shù)，
????????用于實現(xiàn)打開文件瀏覽項
????????:return:
????????'''
????????im_path?=?QFileDialog.getOpenFileName(self,?os.getcwd(),?'打開圖片',
??????????????????????????????????????????????'Image?File(*.jpg);;Image?File(*.png);;Image?File(*.jpeg)')
????????self.input_image_path.setText(im_path[0])

????def?generate_btn_click(self):
????????'''
????????generate_btn按鈕綁定的槽函數(shù)，
????????用于啟動業(yè)務(wù)子線程
????????:return:
????????'''
????????self.thread_.start()
????????self.generate_btn.setEnabled(False)

????def?finished(self,finished):
????????'''
????????接收子線程中finished變量，
????????判定子線程業(yè)務(wù)是否執(zhí)行完成，若執(zhí)行完成則將按鈕狀態(tài)改變?yōu)榭牲c擊狀態(tài)
????????:param?finished:
????????:return:
????????'''
????????if?finished?is?True:
????????????self.generate_btn.setEnabled(True)

class?WorkThread(QThread):
????finished?=?pyqtSignal(bool)

????def?__init__(self,?parent=None):
????????'''
????????子線程初始化函數(shù)
????????:param?parent:
????????'''
????????super(WorkThread,?self).__init__(parent)
????????self.parent?=?parent
????????self.working?=?True

????def?__del__(self):
????????'''
????????子線程停止函數(shù)
????????:return:
????????'''
????????self.working?=?False
????????self.wait()

????def?run(self):
????????'''
????????子線程執(zhí)行函數(shù)
????????:return:
????????'''
????????try:
????????????input_picture_name?=?os.path.basename(self.parent.input_image_path.text().strip())
????????????logger.info(input_picture_name)
????????????output_picture_name?=?'cartoon_'?+?input_picture_name
????????????num_down?=?2??#?縮減像素采樣的數(shù)目
????????????num_bilateral?=?7??#?定義雙邊濾波的數(shù)目
????????????img_rgb?=?cv2.imread(input_picture_name)??#?讀取圖片
????????????#?用高斯金字塔降低取樣
????????????img_color?=?img_rgb

????????????for?_?in?range(num_down):
????????????????img_color?=?cv2.pyrDown(img_color)

????????????#?重復(fù)使用小的雙邊濾波代替一個大的濾波
????????????for?_?in?range(num_bilateral):
????????????????img_color?=?cv2.bilateralFilter(img_color,?d=9,?sigmaColor=9,?sigmaSpace=7)

????????????#?升采樣圖片到原始大小
????????????for?_?in?range(num_down):
????????????????img_color?=?cv2.pyrUp(img_color)

????????????#?轉(zhuǎn)換為灰度并且使其產(chǎn)生中等的模糊
????????????img_gray?=?cv2.cvtColor(img_color,?cv2.COLOR_RGB2GRAY)
????????????img_blur?=?cv2.medianBlur(img_gray,?7)
????????????#?檢測到邊緣并且增強其效果
????????????img_edge?=?cv2.adaptiveThreshold(img_blur,?255,
?????????????????????????????????????????????cv2.ADAPTIVE_THRESH_MEAN_C,
?????????????????????????????????????????????cv2.THRESH_BINARY,
?????????????????????????????????????????????blockSize=9,
?????????????????????????????????????????????C=2)
????????????#?轉(zhuǎn)換回彩色圖像
????????????img_edge?=?cv2.cvtColor(img_edge,?cv2.COLOR_GRAY2RGB)
????????????img_cartoon?=?cv2.bitwise_and(img_color,?img_edge)
????????????#?保存轉(zhuǎn)換后的圖片
????????????cv2.imwrite(output_picture_name,?img_cartoon)

????????????logger.info('動漫圖片轉(zhuǎn)換完成！')
????????????self.finished.emit(True)
????????except?Exception?as?e:
????????????logger.error(e)

if?__name__?==?'__main__':
????app?=?QApplication(sys.argv)
????main?=?CartoonUI()
????main.show()
????sys.exit(app.exec_())