在視覺設計領域中,設計師們每每會由於一些簡單需求付出至關多的時間,好比修改文案內容,設計簡單的海報版式,針對不一樣機型、展位的多尺寸拓展等。這些工做須要耗費大量的時間、人力成本(5~6張/人日),但對設計師的進步成長起到的做用卻很是有限。另外一方面,精準營銷是將來的大趨勢,在大流量背景下,首頁的海報資源展位須要展現「千人千面」的效果,這對海報的生產效率也提出了很是高的要求。因此,咱們美團外賣技術團隊嘗試結合AI技術,來協助設計師避免這種低收益、高重複的任務,同時低成本、高效率、高質量地完成海報圖片的生成。本文以Banner(橫版海報)爲例,介紹咱們在海報設計與AI技術結合方面所進行的一些探索和研究。算法
什麼是Banner的設計過程?咱們嘗試總結了對Banner設計的理解,Banner的設計過程是一系列的具有某種特徵屬性的素材圖層的有序疊加過程。這裏的特徵屬性既包括顏色、形狀、紋理、主題等視覺屬性,也包括位置、大小、貼邊等空間屬性。在這個過程當中,哪些環節能夠被機器算法所探索呢?文獻[1]研究瞭如何調整圖像的顏色分佈,使雜誌封面的視覺效果更加符合人眼的視覺特性;文獻[2]以此爲基礎,引入了基於顯著性識別的圖像裁剪,並使用優化方法來解決佈局問題。阿里巴巴的鹿班系統在去年雙十一當天,生成1.7億張Banner;京東內部也在孵化玲瓏和莎士比亞系統,更加智能地設計文案和Banner。網絡
圖1 封面配色&佈局設計[2]框架
在設計領域的一些子問題上,能夠用算法來挖掘出數據背後的規律(如圖1所示)。那麼,可否構建一個完整的學習算法和處理系統,統一解決Banner設計中全部的子問題(配色、佈局、搭配、生成)呢?函數
素材圖層是Banner的基礎元素,其自己能夠被特徵化,同時組成Banner的若干元素間的疊加順序能夠被序列化,所以,算法實際是在學習「在何時,選擇某種素材,放在哪裏」。佈局
圖2 流程框架學習
如圖2所示,爲了解決這個問題,咱們設計規劃器、優化器、生成器來共同構建海報設計的學習與生產過程。其中:優化
如何提取素材圖片的特徵屬性,這是比較典型的分類問題。在計算機視覺領域,傳統方案是提取圖像的顏色、梯度等低級語義特徵[3],結合傳統的分類器(LR、SVM等)來實現分類。近年來,基於深度學習的方法由於能表達更爲複雜的語義特徵,逐漸成爲主流方法[4]。如圖3所示咱們提取傳統的低級語義特徵,以及基於CNN的高級語義特徵,來共同完成素材特徵屬性提取。 ui
圖3 素材庫-特徵提取設計
完成素材的數據化工做後,怎樣學習Banner的設計過程?3d
做爲一種生成模型,對抗生成網絡(GAN)[5]在近年廣爲應用,其優點是能夠端到端地訓練圖像生成模型,但在咱們的應用場景下,GAN存在如下兩個問題:
在上文中有提到,Banner設計過程是素材圖層依次疊加的過程。所以,咱們能夠用序列生成模型來擬合這個過程[6]。在建模過程當中,咱們把素材視做詞彙(Word),海報視做句子(Sentence),詞彙索引視爲離散化的特徵索引,素材疊加順序就能夠視爲句子中的詞順序[7]。
圖4 規劃器-序列生成
圖4是咱們使用的序列生成模型,輸入主體信息和目標風格,輸出素材特徵的序列。爲了加強預測過程當中多條路徑結果的多樣性,咱們在監督性地訓練每一個時刻的輸出以外,還引入了評估整個序列合理性的Object loss。如圖5所示,借鑑SeqGAN的思想,Object loss能夠由判別器來提供[8]。
圖5 SeqGAN[8]
規劃器預測素材的量化特徵,爲了確保最終成圖符合美學標準,須要一個後處理的過程(圖6)。咱們用優化器來解決這個問題。從本質上講,這是一個優化過程。經過和設計師們的溝通,咱們設計了一些基於常規設計理念和美學標準的目標函數,動做集合包括移動、縮放、亮度調整等,結合優化方法,提高Banner的視覺效果。
圖6 優化器
優化後的素材特徵序列,經過生成器來渲染成圖。如圖7所示,對於素材庫檢索不到符合某種特徵屬性的素材的狀況,咱們設計了圖像風格遷移的方法來實現圖像特徵遷移。這裏的特徵能夠是顏色、形狀、紋理等低級特徵,也能夠是某種語義化的視覺風格特徵,對後者來講,能夠將源圖像的內容Content和目標圖像的風格Style在某種特徵空間(CNN中的某一層)裏作融合,實現風格遷移[9,10]。
圖7 素材生成
「千人千面」的精準營銷是將來營銷策略的趨勢,這對商品素材的豐富性提出了很是高的要求;從爲商家賦能的角度來講,也須要爲商家提供更多樣的海報版式,這也要求系統具有海報風格的快速學習能力和拓展能力。對此,在常規設計風格的研究以外,咱們從如下3個方面作了一些拓展研究。
商品素材的豐富度與美學質量是精細化營銷及海報美學質量很是重要的一環。其中最核心的要求是圖像分割的能力[11,12]。以全卷積網絡(FCN)爲基礎,如圖8所示,咱們採起如下幾個在目標分割領域常見的技巧,來實現商品圖片的目標分割:
圖8 圖像語義分割&摳圖(結構圖部分參考DeepLab v3+[12])
這種基於語義分割方法的結果,在專業設計師人工評審質量的過程當中,發現主體邊緣有時會出現明顯的鋸齒感。通過分析,咱們認爲有如下兩個緣由:
爲此,如圖8所示,咱們在圖像分割的輸出結果以外,結合了Image Matting方法:
圖9 商品主體摳圖
另外,基於圖像美學質量評分模型,咱們會優先選擇質量分高的圖片做爲主體素材來源。對中低分的圖片,將來能夠考慮借鑑Cycle-GAN[13]的思想,設計基於半監督和GAN的圖像加強網絡,美化後再通過圖像分割產生主體素材。
上述的常規設計風格的學習屬於一種廣義的設計風格,須要設計師先期投入不少精力作風格劃分以及數據收集、處理。爲了快速適配熱點場景,咱們借鑑圖像檢索技術(如圖10所示),提取素材圖片的CNN特徵及顏色特徵,使用歐式距離來度量素材類似度,這能節省人工打標籤的成本,實現基於固定模板的自動拓展及生成(海報臨摹)。
圖10 素材圖像檢索與模板拓展
在平常工做中,設計師在設計出Banner後,每每要花費很長時間對不一樣展位、不一樣版本、不一樣機型作多尺寸適配(如圖11所示)。可否用算法來協助人工提效?在素材已經肯定,而且相互的位置關係近乎肯定的條件下作多分辨率適配,這本質上是一個優化問題,咱們在上文佈局優化器的基礎上,增長元素的局部相對位置與全局絕對位置的拓撲關係做爲目標函數。目前,系統支持在某個固定寬高比±30%範圍內的任意分辨率適配,將來會進一步擴大適配範圍。
圖11 多分辨率拓展
目前,咱們的Banner智能設計系統爲鑽展(外賣首頁廣告位)、商家店鋪裝修等業務提供穩定的設計能力;素材加工等子能力也在爲外賣、閃購等商品圖片提供技術支持。後續咱們會從擴展常規設計風格、語義相關的顏色及素材挖掘、自動解析數據、構建自評估學習閉環等方面繼續研究,進一步提升算法的設計能力和適用性,儘量協助設計師提升效率,下降高重複性工做的時間和經濟成本。
參考文獻
[1] A. Jahanian, J. Liu, D. Tretter, Q. Lin, E. O’Brien-Strain, S. Lee, N. Lyons, and J. P. Allebach. "Automatic Design of Colors for Magazine Covers". In Proc. IS&T/SPIE Electronic Imaging, International Society for Optics and Photonics, 2013 [2] X. Y. Yang, T. Mei, Y. Q. Xu, Y. Rui, S. P. Li. "Automatic Generation of Visual-Textual Presentation Layout". ACM Transactions on Multimedia Computing, Communications, and Applications, 2017 [3] David G. Lowe. "Distinctive Image Features from Scale-Invariant Keypoints". International journal of computer vision, 2004 [4] Alex Krizhevsky, Ilya Sutskever, Geoffrey E. Hinton. "ImageNet Classification with Deep Convolutional Neural Networks". NIPS, 2012 [5] I. Goodfellow, J. Pouget-Abadie, M. Mirza, B. Xu, D. Warde-Farley, S. Ozair, A. Courville, Y. Bengio. "Generative Adversarial Networks". NIPS, 2014 [6] K. Kawakami. "Supervised Sequence Labelling with Recurrent Neural Networks". Studies in Computational Intelligence, 2008 [7] T. Mikolov. "Statistical Language Models based on Neural Networks". 2012 [8] L. Yu, W. Zhang, J. Wang, Y. Yu. "SeqGAN: Sequence Generative Adversarial Nets with Policy Gradient". AAAI, 2017 [9] L.A. Gatys, A.S. Ecker, M. Bethge. "Image Style Transfer Using Convolutional Neural Networks". CVPR, 2016 [10] Y. Li, M.Y. Liu, X. Li, M.H. Yang, J. Kautz. "A Closed-form Solution to Photorealistic Image Stylization". ECCV, 2018 [11] J. Long, E. Shelhamer, T. Darrell. "Fully Convolutional Networks for Semantic Segmentation". CVPR, 2015 [12] L.C. Chen, Y. Zhu, G. Papandreou, F. Schroff, H. Adam. "Encoder-Decoder with Atrous Separable Convolution for Semantic Image Segmentation". ECCV, 2018 [13] J.Y. Zhu, T. Park, P. Isola, A. A. Efros. "Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks". ICCV, 2017
曉星,2017年6月加入美團,目前主要負責美團外賣圖像內容挖掘、加強、生成方面的相關工做,致力於圖像相關技術的積累及落地。