radames/Text2Human-API

1from PyQt5 import QtCore, QtGui, QtWidgets2from PyQt5.QtCore import *3from PyQt5.QtGui import *4from PyQt5.QtWidgets import *5 6 7class Ui_Form(object):8 9    def setupUi(self, Form):10        Form.setObjectName("Form")11        Form.resize(1250, 670)12 13        self.pushButton_2 = QtWidgets.QPushButton(Form)14        self.pushButton_2.setGeometry(QtCore.QRect(20, 60, 97, 27))15        self.pushButton_2.setObjectName("pushButton_2")16 17        self.pushButton_6 = QtWidgets.QPushButton(Form)18        self.pushButton_6.setGeometry(QtCore.QRect(20, 100, 97, 27))19        self.pushButton_6.setObjectName("pushButton_6")20 21        # Generate Parsing22        self.pushButton_0 = QtWidgets.QPushButton(Form)23        self.pushButton_0.setGeometry(QtCore.QRect(126, 60, 150, 27))24        self.pushButton_0.setObjectName("pushButton_0")25 26        # Generate Human27        self.pushButton_1 = QtWidgets.QPushButton(Form)28        self.pushButton_1.setGeometry(QtCore.QRect(126, 100, 150, 27))29        self.pushButton_1.setObjectName("pushButton_1")30 31        # shape text box32        self.label_heading_1 = QtWidgets.QLabel(Form)33        self.label_heading_1.setText('Describe the shape.')34        self.label_heading_1.setObjectName("label_heading_1")35        self.label_heading_1.setGeometry(QtCore.QRect(320, 20, 200, 20))36 37        self.message_box_1 = QtWidgets.QLineEdit(Form)38        self.message_box_1.setGeometry(QtCore.QRect(320, 50, 256, 80))39        self.message_box_1.setObjectName("message_box_1")40        self.message_box_1.setAlignment(Qt.AlignTop)41 42        # texture text box43        self.label_heading_2 = QtWidgets.QLabel(Form)44        self.label_heading_2.setText('Describe the textures.')45        self.label_heading_2.setObjectName("label_heading_2")46        self.label_heading_2.setGeometry(QtCore.QRect(620, 20, 200, 20))47 48        self.message_box_2 = QtWidgets.QLineEdit(Form)49        self.message_box_2.setGeometry(QtCore.QRect(620, 50, 256, 80))50        self.message_box_2.setObjectName("message_box_2")51        self.message_box_2.setAlignment(Qt.AlignTop)52 53        # title icon54        self.title_icon = QtWidgets.QLabel(Form)55        self.title_icon.setGeometry(QtCore.QRect(30, 10, 200, 50))56        self.title_icon.setPixmap(57            QtGui.QPixmap('./ui/icons/icon_title.png').scaledToWidth(200))58 59        # palette icon60        self.palette_icon = QtWidgets.QLabel(Form)61        self.palette_icon.setGeometry(QtCore.QRect(950, 10, 256, 128))62        self.palette_icon.setPixmap(63            QtGui.QPixmap('./ui/icons/icon_palette.png').scaledToWidth(256))64 65        # top66        self.pushButton_8 = QtWidgets.QPushButton('   top', Form)67        self.pushButton_8.setGeometry(QtCore.QRect(940, 120, 120, 27))68        self.pushButton_8.setObjectName("pushButton_8")69        self.pushButton_8.setStyleSheet(70            "text-align: left; padding-left: 10px;")71        self.pushButton_8.setIcon(QIcon('./ui/color_blocks/class_top.png'))72        # skin73        self.pushButton_9 = QtWidgets.QPushButton('   skin', Form)74        self.pushButton_9.setGeometry(QtCore.QRect(940, 165, 120, 27))75        self.pushButton_9.setObjectName("pushButton_9")76        self.pushButton_9.setStyleSheet(77            "text-align: left; padding-left: 10px;")78        self.pushButton_9.setIcon(QIcon('./ui/color_blocks/class_skin.png'))79        # outer80        self.pushButton_10 = QtWidgets.QPushButton('   outer', Form)81        self.pushButton_10.setGeometry(QtCore.QRect(940, 210, 120, 27))82        self.pushButton_10.setObjectName("pushButton_10")83        self.pushButton_10.setStyleSheet(84            "text-align: left; padding-left: 10px;")85        self.pushButton_10.setIcon(QIcon('./ui/color_blocks/class_outer.png'))86        # face87        self.pushButton_11 = QtWidgets.QPushButton('   face', Form)88        self.pushButton_11.setGeometry(QtCore.QRect(940, 255, 120, 27))89        self.pushButton_11.setObjectName("pushButton_11")90        self.pushButton_11.setStyleSheet(91            "text-align: left; padding-left: 10px;")92        self.pushButton_11.setIcon(QIcon('./ui/color_blocks/class_face.png'))93        # skirt94        self.pushButton_12 = QtWidgets.QPushButton('   skirt', Form)95        self.pushButton_12.setGeometry(QtCore.QRect(940, 300, 120, 27))96        self.pushButton_12.setObjectName("pushButton_12")97        self.pushButton_12.setStyleSheet(98            "text-align: left; padding-left: 10px;")99        self.pushButton_12.setIcon(QIcon('./ui/color_blocks/class_skirt.png'))100        # hair101        self.pushButton_13 = QtWidgets.QPushButton('   hair', Form)102        self.pushButton_13.setGeometry(QtCore.QRect(940, 345, 120, 27))103        self.pushButton_13.setObjectName("pushButton_13")104        self.pushButton_13.setStyleSheet(105            "text-align: left; padding-left: 10px;")106        self.pushButton_13.setIcon(QIcon('./ui/color_blocks/class_hair.png'))107        # dress108        self.pushButton_14 = QtWidgets.QPushButton('   dress', Form)109        self.pushButton_14.setGeometry(QtCore.QRect(940, 390, 120, 27))110        self.pushButton_14.setObjectName("pushButton_14")111        self.pushButton_14.setStyleSheet(112            "text-align: left; padding-left: 10px;")113        self.pushButton_14.setIcon(QIcon('./ui/color_blocks/class_dress.png'))114        # headwear115        self.pushButton_15 = QtWidgets.QPushButton('   headwear', Form)116        self.pushButton_15.setGeometry(QtCore.QRect(940, 435, 120, 27))117        self.pushButton_15.setObjectName("pushButton_15")118        self.pushButton_15.setStyleSheet(119            "text-align: left; padding-left: 10px;")120        self.pushButton_15.setIcon(121            QIcon('./ui/color_blocks/class_headwear.png'))122        # pants123        self.pushButton_16 = QtWidgets.QPushButton('   pants', Form)124        self.pushButton_16.setGeometry(QtCore.QRect(940, 480, 120, 27))125        self.pushButton_16.setObjectName("pushButton_16")126        self.pushButton_16.setStyleSheet(127            "text-align: left; padding-left: 10px;")128        self.pushButton_16.setIcon(QIcon('./ui/color_blocks/class_pants.png'))129        # eyeglasses130        self.pushButton_17 = QtWidgets.QPushButton('   eyeglass', Form)131        self.pushButton_17.setGeometry(QtCore.QRect(940, 525, 120, 27))132        self.pushButton_17.setObjectName("pushButton_17")133        self.pushButton_17.setStyleSheet(134            "text-align: left; padding-left: 10px;")135        self.pushButton_17.setIcon(136            QIcon('./ui/color_blocks/class_eyeglass.png'))137        # rompers138        self.pushButton_18 = QtWidgets.QPushButton('   rompers', Form)139        self.pushButton_18.setGeometry(QtCore.QRect(940, 570, 120, 27))140        self.pushButton_18.setObjectName("pushButton_18")141        self.pushButton_18.setStyleSheet(142            "text-align: left; padding-left: 10px;")143        self.pushButton_18.setIcon(144            QIcon('./ui/color_blocks/class_rompers.png'))145        # footwear146        self.pushButton_19 = QtWidgets.QPushButton('   footwear', Form)147        self.pushButton_19.setGeometry(QtCore.QRect(940, 615, 120, 27))148        self.pushButton_19.setObjectName("pushButton_19")149        self.pushButton_19.setStyleSheet(150            "text-align: left; padding-left: 10px;")151        self.pushButton_19.setIcon(152            QIcon('./ui/color_blocks/class_footwear.png'))153 154        # leggings155        self.pushButton_20 = QtWidgets.QPushButton('   leggings', Form)156        self.pushButton_20.setGeometry(QtCore.QRect(1100, 120, 120, 27))157        self.pushButton_20.setObjectName("pushButton_10")158        self.pushButton_20.setStyleSheet(159            "text-align: left; padding-left: 10px;")160        self.pushButton_20.setIcon(161            QIcon('./ui/color_blocks/class_leggings.png'))162 163        # ring164        self.pushButton_21 = QtWidgets.QPushButton('   ring', Form)165        self.pushButton_21.setGeometry(QtCore.QRect(1100, 165, 120, 27))166        self.pushButton_21.setObjectName("pushButton_2`0`")167        self.pushButton_21.setStyleSheet(168            "text-align: left; padding-left: 10px;")169        self.pushButton_21.setIcon(QIcon('./ui/color_blocks/class_ring.png'))170 171        # belt172        self.pushButton_22 = QtWidgets.QPushButton('   belt', Form)173        self.pushButton_22.setGeometry(QtCore.QRect(1100, 210, 120, 27))174        self.pushButton_22.setObjectName("pushButton_2`0`")175        self.pushButton_22.setStyleSheet(176            "text-align: left; padding-left: 10px;")177        self.pushButton_22.setIcon(QIcon('./ui/color_blocks/class_belt.png'))178 179        # neckwear180        self.pushButton_23 = QtWidgets.QPushButton('   neckwear', Form)181        self.pushButton_23.setGeometry(QtCore.QRect(1100, 255, 120, 27))182        self.pushButton_23.setObjectName("pushButton_2`0`")183        self.pushButton_23.setStyleSheet(184            "text-align: left; padding-left: 10px;")185        self.pushButton_23.setIcon(186            QIcon('./ui/color_blocks/class_neckwear.png'))187 188        # wrist189        self.pushButton_24 = QtWidgets.QPushButton('   wrist', Form)190        self.pushButton_24.setGeometry(QtCore.QRect(1100, 300, 120, 27))191        self.pushButton_24.setObjectName("pushButton_2`0`")192        self.pushButton_24.setStyleSheet(193            "text-align: left; padding-left: 10px;")194        self.pushButton_24.setIcon(QIcon('./ui/color_blocks/class_wrist.png'))195 196        # socks197        self.pushButton_25 = QtWidgets.QPushButton('   socks', Form)198        self.pushButton_25.setGeometry(QtCore.QRect(1100, 345, 120, 27))199        self.pushButton_25.setObjectName("pushButton_2`0`")200        self.pushButton_25.setStyleSheet(201            "text-align: left; padding-left: 10px;")202        self.pushButton_25.setIcon(QIcon('./ui/color_blocks/class_socks.png'))203 204        # tie205        self.pushButton_26 = QtWidgets.QPushButton('   tie', Form)206        self.pushButton_26.setGeometry(QtCore.QRect(1100, 390, 120, 27))207        self.pushButton_26.setObjectName("pushButton_2`0`")208        self.pushButton_26.setStyleSheet(209            "text-align: left; padding-left: 10px;")210        self.pushButton_26.setIcon(QIcon('./ui/color_blocks/class_tie.png'))211 212        # earstuds213        self.pushButton_27 = QtWidgets.QPushButton('   necklace', Form)214        self.pushButton_27.setGeometry(QtCore.QRect(1100, 435, 120, 27))215        self.pushButton_27.setObjectName("pushButton_2`0`")216        self.pushButton_27.setStyleSheet(217            "text-align: left; padding-left: 10px;")218        self.pushButton_27.setIcon(219            QIcon('./ui/color_blocks/class_necklace.png'))220 221        # necklace222        self.pushButton_28 = QtWidgets.QPushButton('   earstuds', Form)223        self.pushButton_28.setGeometry(QtCore.QRect(1100, 480, 120, 27))224        self.pushButton_28.setObjectName("pushButton_2`0`")225        self.pushButton_28.setStyleSheet(226            "text-align: left; padding-left: 10px;")227        self.pushButton_28.setIcon(228            QIcon('./ui/color_blocks/class_earstuds.png'))229 230        # bag231        self.pushButton_29 = QtWidgets.QPushButton('   bag', Form)232        self.pushButton_29.setGeometry(QtCore.QRect(1100, 525, 120, 27))233        self.pushButton_29.setObjectName("pushButton_2`0`")234        self.pushButton_29.setStyleSheet(235            "text-align: left; padding-left: 10px;")236        self.pushButton_29.setIcon(QIcon('./ui/color_blocks/class_bag.png'))237 238        # glove239        self.pushButton_30 = QtWidgets.QPushButton('   glove', Form)240        self.pushButton_30.setGeometry(QtCore.QRect(1100, 570, 120, 27))241        self.pushButton_30.setObjectName("pushButton_2`0`")242        self.pushButton_30.setStyleSheet(243            "text-align: left; padding-left: 10px;")244        self.pushButton_30.setIcon(QIcon('./ui/color_blocks/class_glove.png'))245 246        # background247        self.pushButton_31 = QtWidgets.QPushButton('   background', Form)248        self.pushButton_31.setGeometry(QtCore.QRect(1100, 615, 120, 27))249        self.pushButton_31.setObjectName("pushButton_2`0`")250        self.pushButton_31.setStyleSheet(251            "text-align: left; padding-left: 10px;")252        self.pushButton_31.setIcon(QIcon('./ui/color_blocks/class_bg.png'))253 254        self.graphicsView = QtWidgets.QGraphicsView(Form)255        self.graphicsView.setGeometry(QtCore.QRect(20, 140, 256, 512))256        self.graphicsView.setObjectName("graphicsView")257        self.graphicsView_2 = QtWidgets.QGraphicsView(Form)258        self.graphicsView_2.setGeometry(QtCore.QRect(320, 140, 256, 512))259        self.graphicsView_2.setObjectName("graphicsView_2")260        self.graphicsView_3 = QtWidgets.QGraphicsView(Form)261        self.graphicsView_3.setGeometry(QtCore.QRect(620, 140, 256, 512))262        self.graphicsView_3.setObjectName("graphicsView_3")263 264        self.retranslateUi(Form)265        self.pushButton_2.clicked.connect(Form.open_densepose)266        self.pushButton_6.clicked.connect(Form.save_img)267        self.pushButton_8.clicked.connect(Form.top_mode)268        self.pushButton_9.clicked.connect(Form.skin_mode)269        self.pushButton_10.clicked.connect(Form.outer_mode)270        self.pushButton_11.clicked.connect(Form.face_mode)271        self.pushButton_12.clicked.connect(Form.skirt_mode)272        self.pushButton_13.clicked.connect(Form.hair_mode)273        self.pushButton_14.clicked.connect(Form.dress_mode)274        self.pushButton_15.clicked.connect(Form.headwear_mode)275        self.pushButton_16.clicked.connect(Form.pants_mode)276        self.pushButton_17.clicked.connect(Form.eyeglass_mode)277        self.pushButton_18.clicked.connect(Form.rompers_mode)278        self.pushButton_19.clicked.connect(Form.footwear_mode)279        self.pushButton_20.clicked.connect(Form.leggings_mode)280        self.pushButton_21.clicked.connect(Form.ring_mode)281        self.pushButton_22.clicked.connect(Form.belt_mode)282        self.pushButton_23.clicked.connect(Form.neckwear_mode)283        self.pushButton_24.clicked.connect(Form.wrist_mode)284        self.pushButton_25.clicked.connect(Form.socks_mode)285        self.pushButton_26.clicked.connect(Form.tie_mode)286        self.pushButton_27.clicked.connect(Form.earstuds_mode)287        self.pushButton_28.clicked.connect(Form.necklace_mode)288        self.pushButton_29.clicked.connect(Form.bag_mode)289        self.pushButton_30.clicked.connect(Form.glove_mode)290        self.pushButton_31.clicked.connect(Form.background_mode)291        self.pushButton_0.clicked.connect(Form.generate_parsing)292        self.pushButton_1.clicked.connect(Form.generate_human)293 294        QtCore.QMetaObject.connectSlotsByName(Form)295 296    def retranslateUi(self, Form):297        _translate = QtCore.QCoreApplication.translate298        Form.setWindowTitle(_translate("Form", "Text2Human"))299        self.pushButton_2.setText(_translate("Form", "Load Pose"))300        self.pushButton_6.setText(_translate("Form", "Save Image"))301 302        self.pushButton_0.setText(_translate("Form", "Generate Parsing"))303        self.pushButton_1.setText(_translate("Form", "Generate Human"))304 305 306if __name__ == "__main__":307    import sys308    app = QtWidgets.QApplication(sys.argv)309    Form = QtWidgets.QWidget()310    ui = Ui_Form()311    ui.setupUi(Form)312    Form.show()313    sys.exit(app.exec_())314