pip install git
git clone https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix

pip install -r requirements.txt

python test.py --dataroot ./datasets/edges2shoes --name edges2shoes_pretrained --model pix2pix --direction BtoA

pip install PyQt5

jupyter notebook

self.setWindowTitle("Paint with PyQt5")
merge = 40
size = canvas_size - merge
self.setGeometry(merge, merge, size, size)

self.image = QImage(self.size(), QImage.Format_RGB32) 
self.image.fill(Qt.white)

self.drawing = False 
self.brushSize = brush_size
self.brushColor = Qt.black
self.lastPoint = QPoint()

mainMenu = self.menuBar() 
fileMenu = mainMenu.addMenu("File")

saveAction = QAction("Save", self) 
saveAction.setShortcut(QKeySequence('Ctrl+s'))
fileMenu.addAction(saveAction)
saveAction.triggered.connect(self.save)

clearAction = QAction("Clear", self) 
clearAction.setShortcut(QKeySequence('Ctrl+c')) 
fileMenu.addAction(clearAction) 
clearAction.triggered.connect(self.clear)

exitAction = QAction("Exit", self)
exitAction.setShortcut(QKeySequence('Ctrl+q'))
fileMenu.addAction(exitAction)
exitAction.triggered.connect(self.exitapp)

    def mousePressEvent(self, event): 
        if event.button() == Qt.LeftButton: 
            self.drawing = True
            self.lastPoint = event.pos()

   def mouseMoveEvent(self, event): 
          
        if (event.buttons() & Qt.LeftButton) & self.drawing: 
              
            painter = QPainter(self.image) 
            painter.setPen(QPen(self.brushColor, self.brushSize,  
                            Qt.SolidLine, Qt.RoundCap, Qt.RoundJoin)) 
              
            painter.drawLine(self.lastPoint, event.pos()) 
              
            self.lastPoint = event.pos() 

            self.update()

    def mouseReleaseEvent(self, event): 
  
        if event.button() == Qt.LeftButton:
            self.drawing = False

    # 建立 painter 後還須建立畫圖的事件 
    def paintEvent(self, event): 
        canvasPainter = QPainter(self) 
        canvasPainter.drawImage(self.rect(), self.image, self.image.rect()) 
   
    # 清除畫布
    def clear(self): 
        
        self.image.fill(Qt.white)
        self.update() 
    
    # 離開視窗
    def exitapp(self):
        self.close()

    def initImage(self, filepath):

        trg_size = 256

        new_img = np.zeros([256,512,3], dtype=np.uint8)

        new_img.fill(255)

        org_img = cv2.imread('{}'.format(filepath))
        src_img = cv2.resize(org_img, (trg_size, trg_size) )

        print('org_img {} > src_img {}'.format(org_img.shape, src_img.shape))

        for h in range(trg_size):
            for w in range(trg_size*2):
                if w< 256:
                    new_img[h][w] = src_img[h][w]
                else:
                    break

        cv2.imwrite(filepath, new_img)

        # 儲存畫布
    def save(self):
        filePath = {your path}

        self.image.save(filePath) 
        self.initImage(filePath)
        print('save image: ', filePath)
        self.close()     

# 實例化 app 這支程式
if not QtWidgets.QApplication.instance():
    app = QtWidgets.QApplication(sys.argv)
else:
    app = QtWidgets.QApplication.instance() 

# 獲得螢幕可使用範圍
screen = app.primaryScreen()
rect = screen.availableGeometry()
print('Available: %d x %d' % (rect.width(), rect.height()))

# 宣告畫布大小以及筆刷大小
canvas_size = rect.width() if rect.width() < rect.height() else rect.height()
brush_size = 10

# 實例化視窗
window = Window(canvas_size, brush_size) 
  
# 顯示視窗
window.show() 
  
# 執行 app
sys.exit(app.exec())

!python test.py --dataroot datasets\edges2shoes --model pix2pix --name edges2shoes_pretrained
        
ImPath =  {your path} + ‘custom_edge_fake_B.png’
im = cv2.imread(ImPath)
cv2.imshow('result', im)
cv2.waitKey(0)
cv2.destroyAllWindows()