Initial commit

Author: Robin-Huang

.idea/misc.xml

@@ -0,0 +1,11 @@
+import sys
+import serverWindow
+from PyQt5.QtWidgets import QApplication
+
+if __name__ == '__main__':
+    app = QApplication(sys.argv)
+
+    server = serverWindow.serverWindow()
+    server.show()
+
+    sys.exit(app.exec())

.idea/modules.xml

@@ -0,0 +1,16 @@
+大紅紋鳳蝶
+無尾鳳蝶
+柑橘鳳蝶
+青斑鳳蝶
+無尾白紋鳳蝶
+大鳳蝶(雄)
+白紋鳳蝶
+紅紋鳳蝶
+青帶鳳蝶
+玉帶鳳蝶(雄)
+黑鳳蝶
+大鳳蝶(雌)
+玉帶鳳蝶(雌)
+烏鴉鳳蝶
+綠斑鳳蝶
+大琉璃紋鳳蝶

.idea/pyServerPailio.iml

@@ -0,0 +1,16 @@
+byasa polyeuctes termessus
+papilio demoleus
+papilio xuthus
+graphium doson postianus
+papilio castor formosanus
+papilio memnon heronus male
+papilio helenus fortunius
+pachliopta aristolochiae interpositus
+graphium sarpedon connectens
+papilio polytes polytes male
+papilio protenor
+papilio memnon heronus female
+papilio polytes polytes female
+papilio bianor thrasymedes
+graphium agamemnon
+papilio paris nakaharai

.idea/vcs.xml

@@ -0,0 +1,39 @@
+import tensorflow as tf
+
+class PailioRecog(object):
+    def __init__(self):
+        self.graph_file = 'models/papilionidae_graph_4000.pb'
+        self.label_file = 'models/papilionidae_chinese_labels.txt'
+
+    def load_model(self):
+        # Load label data
+        self.labels = []
+        for label in tf.gfile.GFile(self.label_file):
+            self.labels.append(label.rstrip())
+
+        # Load cnn model
+        with tf.gfile.FastGFile(self.graph_file, 'rb') as f:
+            graph_def = tf.GraphDef()
+            graph_def.ParseFromString(f.read())
+            tf.import_graph_def(graph_def, name='')
+
+    def start_recognition(self, image_file):
+
+        self.image = tf.gfile.FastGFile(image_file, 'rb').read() # read image
+
+        with tf.Session() as sess:
+            softmax_tensor = sess.graph.get_tensor_by_name('final_result:0')
+            predict = sess.run(softmax_tensor, {'DecodeJpeg/contents:0': self.image})
+
+            # Sort the label by probability
+            top = predict[0].argsort()[-len(predict[0]):][::-1]
+
+            # for index in top:
+            #     human_string = labels[index]
+            #     score = predict[0][index]
+            #     print(human_string, score)
+
+            self.recogClass = self.labels[top[0]]
+            self.top_score = predict[0][top[0]]
+            # second_score = predict[0][1]
+

main.py

@@ -0,0 +1,130 @@
+# TcpServer
+from PyQt5.QtWidgets import QMainWindow
+from PyQt5.QtCore    import QObject, QFile, QDateTime, QDataStream, QDir, Qt, QIODevice, QByteArray
+from PyQt5.QtGui     import QImage, QPixmap
+from PyQt5.QtNetwork import QTcpServer, QTcpSocket, QHostAddress
+
+from ui_serverWindow import Ui_serverWindow
+from pailio_recognition import PailioRecog
+
+class serverWindow(QMainWindow):
+    def __init__(self, parent = None):
+        super(serverWindow, self).__init__(parent)
+        self.ui      = Ui_serverWindow()
+
+        # Server & Client
+        self.client  = QTcpSocket(self)
+        self.server  = QTcpServer(self)
+
+        # Recieve File
+        self.curTime       = QDateTime()
+        self.file          = QFile()
+        self.image         = QImage()
+        self.bytesReceived = 0
+        self.fileSize      = 0
+        self.filePath      = ''
+        self.savePath      = ''
+
+        self.ui.setupUi(self)
+        self.dirSetting()
+
+        # Pailio Recognition
+        self.ui.stateBrowser.append('準備辨識系統模型...')
+        self.recogPailio = PailioRecog()
+        self.recogPailio.load_model()
+        self.ui.stateBrowser.append('模型準備完成...\n')
+
+
+        # Connect Signal & Slot
+        self.ui.startBtn.clicked.connect(self.startServer)
+        self.ui.stopBtn.clicked.connect(self.stopServer)
+        self.server.newConnection.connect(self.acceptConnection, Qt.QueuedConnection)
+
+    def dirSetting(self):
+        dir = QDir()
+        self.savePath = dir.currentPath() + '/recieve/'
+        self.ui.savePath.setText(self.savePath)
+
+    def startServer(self):
+        self.server.listen(QHostAddress.Any, 37)
+        self.ui.stateBrowser.append('伺服器啟動...')
+        self.ui.startBtn.setEnabled(False)
+        self.ui.stopBtn.setEnabled(True)
+
+    def stopServer(self):
+        self.server.close()
+        self.ui.stateBrowser.append('伺服器關閉...')
+        self.ui.startBtn.setEnabled(True)
+        self.ui.stopBtn.setEnabled(False)
+
+    def acceptConnection(self):
+        self.curTime = QDateTime.currentDateTime()
+        self.filePath = self.savePath + self.curTime.toString('yyyy-MM-dd_hh-mm-ss') + '.jpg'
+        self.file = QFile(self.filePath)
+
+        # Check File
+        if self.file.open(QIODevice.WriteOnly) == 0:
+            self.ui.stateBrowser.append('無法開啟檔案: ' + self.filePath)
+            return
+
+        self.bytesReceived = 0
+        self.fileSize      = 0
+        self.client = self.server.nextPendingConnection()
+
+        # Connect Signal & Slot
+        self.client.readyRead.connect(self.updateProgress)
+        self.client.error.connect(self.displayError)
+
+    def updateProgress(self):
+        stream = QDataStream(self.client)
+
+        # bytesAvailable > sizeof(qint64) = 8
+        if self.client.bytesAvailable() > 8 and self.fileSize == 0:
+            self.fileSize = int(stream.readInt64()) # stream >> self.fileSize
+            self.bytesReceived += 8
+            self.ui.stateBrowser.append('檔案大小: ' + str(self.fileSize / 1024) + 'KB')
+            self.ui.recvProsBar.setMaximum(self.fileSize)
+        else:
+            self.bytesReceived += int(self.client.bytesAvailable())
+            self.ui.recvProsBar.setValue(self.bytesReceived)
+            self.file.write(self.client.readAll())
+
+            # Finish Receive
+            if self.fileSize == self.bytesReceived:
+                self.file.close()
+                self.ui.stateBrowser.append('檔案已儲存: ' + self.file.fileName())
+
+                self.displayImage()
+                self.sendProsResult()
+
+    def displayImage(self):
+        if self.image.load(self.filePath) == 0:
+            self.ui.imageDisplay.setText('無法顯示圖片')
+        else:
+            self.ui.imageDisplay.setPixmap(QPixmap.fromImage(self.image.scaled(300,300)))
+
+    def sendProsResult(self):
+        self.ui.stateBrowser.append('開始辨識...')
+        self.recogPailio.start_recognition(self.filePath)
+        self.ui.stateBrowser.append('辨識完成...')
+
+        self.prosResult = str(self.recogPailio.recogClass) + ' (相似度' + str(self.recogPailio.top_score * 100) + '%)'
+        self.ui.stateBrowser.append('辨識結果: ' + self.prosResult)
+
+        # Send result
+        byteArray  = QByteArray()
+        byteArray.append(bytes(self.prosResult, encoding = 'UTF-8'))
+        # sendStream = QDataStream(byteArray, QIODevice.WriteOnly)
+        # sendStream.writeString(bytes(self.prosResult, encoding = 'UTF-8'))
+        self.client.write(byteArray)
+        self.ui.stateBrowser.append('送出辨識結果...')
+
+
+    def displayError(self, socketError):
+        if socketError == QTcpSocket.RemoteHostClosedError:
+            self.ui.stateBrowser.append('連結中斷...')
+            self.ui.stateBrowser.append('-------------------------\n')
+        else:
+            self.ui.stateBrowser.append('Network error: ' + self.client.errorString())
+            self.ui.stateBrowser.append('-------------------------\n')
+        self.file = 0