Super Resolution with CoreML (Long Version)

by kenmaz

Slide 1

Slide 1 text

Super Resolution with CoreML Kentaro Matsumae Mercari #tryswiftnyc Sep 5th 2018 try! Swift NYC

Slide 2

Slide 2 text

+1 64 6, 64

Slide 3

Slide 3 text

.BOHB"QQ 1SFWJPVTXPSL ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 4

Slide 4 text

.BOHB"QQ 1SFWJPVTXPSL ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 5

Slide 5 text

.BOHB"QQ 1SFWJPVTXPSL ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 6

Slide 6 text

3FTPMVUJPOWT'JMF4J[F ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 7

Slide 7 text

Y Y 3FTPMVUJPOWT'JMF4J[F ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 8

Slide 8 text

Y Y 3FTPMVUJPOWT'JMF4J[F ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 9

Slide 9 text

Y Y 3FTPMVUJPOWT'JMF4J[F ,# ,# ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 10

Slide 10 text

Y Y 3FTPMVUJPOWT'JMF4J[F ,# ,# ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 11

Slide 11 text

4VQFS3FTPMVUJPO -PX3FTPMVUJPO )JHI3FTPMVUJPO ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 12

Slide 12 text

$PSF.-

Slide 13

Slide 13 text

4VQFS3FTPMVUJPOXJUI$PSF.- 4VQFS 3FTPMVUJPO ,# Y ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 14

Slide 14 text

%FNP

Slide 15

Slide 15 text

4VQFS3FTPMVUJPO.FDIBOJTN

Slide 16

Slide 16 text

*NBHF4VQFS3FTPMVUJPO6TJOH %FFQ$POWPMVUJPOBM/FUXPSLT $IBP%POHFUBM

Slide 17

Slide 17 text

43$// 4VQFS3FTPMVUJPOVTJOH$POWPMVUJPOBM /FVSBM/FUXPSL

Slide 18

Slide 18 text

43$// 4VQFS3FTPMVUJPOVTJOH$POWPMVUJPOBM /FVSBM/FUXPSL

Slide 19

Slide 19 text

,FSOFMBOE$POWPMVUJPO ,FSOFM ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 20

Slide 20 text

,FSOFMBOE$POWPMVUJPO ,FSOFM *OQVU*NBHF ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 21

Slide 21 text

,FSOFMBOE$POWPMVUJPO ,FSOFM *OQVU*NBHF ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 22

Slide 22 text

,FSOFMBOE$POWPMVUJPO ,FSOFM *OQVU*NBHF 0VUQVU*NBHF $POWPMVUJPO ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 23

Slide 23 text

/FVSBM/FUXPSL *OQVU ʜ ʜ 0VUQVU

Slide 24

Slide 24 text

$POWPMVUJPOBM/FVSBM/FUXPSL $// *OQVU ʜ ʜ 0VUQVU

Slide 25

Slide 25 text

$POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 26

Slide 26 text

$POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 27

Slide 27 text

$POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO *OQVU -PX3FTPMVUJPO

Slide 28

Slide 28 text

$POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO *OQVU -PX3FTPMVUJPO

Slide 29

Slide 29 text

$POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO *OQVU -PX3FTPMVUJPO

Slide 30

Slide 30 text

0VUQVU )JHI3FTPMVUJPO $POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO *OQVU -PX3FTPMVUJPO

Slide 31

Slide 31 text

0VUQVU )JHI3FTPMVUJPO *NBHF $POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 32

Slide 32 text

0VUQVU )JHI3FTPMVUJPO *NBHF $POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 33

Slide 33 text

0VUQVU )JHI3FTPMVUJPO *NBHF $POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 34

Slide 34 text

0VUQVU )JHI3FTPMVUJPO *NBHF $POW-BZFS $POW-BZFS $POW-BZFS 4VQFS3FTPMVUJPOXJUI$// 43$// ʜ ʜ ʜ ʜ ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 35

Slide 35 text

5SBJOJOH

Slide 36

Slide 36 text

)BSEUPDPMMFDUUSBJOJOHEBUB

Slide 37

Slide 37 text

&BTZUPDPMMFDUUSBJOJOHEBUBGPS43$// )3*NBHFT ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 38

Slide 38 text

&BTZUPDPMMFDUUSBJOJOHEBUBGPS43$// )3*NBHFT ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 39

Slide 39 text

&BTZUPDPMMFDUUSBJOJOHEBUBGPS43$// )3*NBHFT )31BUDIFT ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 40

Slide 40 text

&BTZUPDPMMFDUUSBJOJOHEBUBGPS43$// )3*NBHFT -31BUDIFT )31BUDIFT Y #JDVCJD *OUFSQPMBUJPO ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 41

Slide 41 text

5SBJOJOH4FU ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 42

Slide 42 text

ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 43

Slide 43 text

5SBJOJOH43$// )31BUDI -31BUDI *OQVU ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 44

Slide 44 text

5SBJOJOH43$// 0VUQVU )31BUDI -31BUDI *OQVU ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 45

Slide 45 text

5SBJOJOH43$// 0VUQVU -PTT 'VODUJPO )31BUDI -31BUDI *OQVU ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 46

Slide 46 text

5SBJOJOH43$// 0VUQVU -PTT 'VODUJPO -PTT )31BUDI -31BUDI *OQVU ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 47

Slide 47 text

5SBJOJOH43$// 0VUQVU -PTT 'VODUJPO 6QEBUFLFSOFMUPNJOJNJ[FMPTT -PTT )31BUDI -31BUDI *OQVU ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 48

Slide 48 text

5SBJOJOH43$// 0VUQVU -PTT 'VODUJPO 6QEBUFLFSOFMUPNJOJNJ[FMPTT )31BUDI -31BUDI *OQVU ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 49

Slide 49 text

,FSOFM8FJHIU CFGPSFUSBJOJOH

Slide 50

Slide 50 text

,FSOFM8FJHIU BGUFSUSBJOJOH

Slide 51

Slide 51 text

,FSOFM8FJHIU BGUFSUSBJOJOH (BVTTJBO'JMUFS

Slide 52

Slide 52 text

,FSOFM8FJHIU BGUFSUSBJOJOH (BVTTJBO'JMUFS -BQMBDJBO'JMUFS -BQMBDJBOpMUFS

Slide 53

Slide 53 text

,FSOFM8FJHIU BGUFSUSBJOJOH (BVTTJBO'JMUFS -BQMBDJBO'JMUFS -BQMBDJBOpMUFS &EHF%FUFDUPS'JMUFS

Slide 54

Slide 54 text

*NQMFNFOUBUJPO43$//

Slide 55

Slide 55 text

43$//.PEFM4USVDUVSF ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 56

Slide 56 text

43$//.PEFM4USVDUVSF < > < > < > < > < > ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 57

Slide 57 text

43$//.PEFM4USVDUVSF < > < > < > < > < > QBSBNFUFST ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 58

Slide 58 text

5SBJOJOHDPEF ch = 1 SRCNN = Sequential(input_shape=(200, 200, ch)) SRCNN.add(Conv2D(128, 9, 9, activation='relu')) SRCNN.add(Conv2D(64, 3, 3, activation='relu')) SRCNN.add(Conv2D(ch, 5, 5, activation='linear')) adam = Adam(lr=0.0003) SRCNN.compile(optimizer=adam, loss='mean_squared_error') SRCNN.fit(..) SRCNN.save(model_dir, 'model.h5')

Slide 59

Slide 59 text

$POWFSUUP$PSF.-.PEFM from coremltools.converters.keras import convert model = convert(‘model.h5’, …) model.save('SRCNN.mlmodel') SRCNN.mlmodel

Slide 60

Slide 60 text

$PSF.-.PEFMpMFTJ[F SRCNN.mlmodel 440 KB IUUQTEFWFMPQFSBQQMFDPNNBDIJOFMFBSOJOHCVJMESVONPEFMT .PCJMF/FU .# 4RVFF[F/FU .# (PPH-F/FU1MBDFT .# 3FTOFU .# *ODFQUJPO7 .# 7(( .#

Slide 61

Slide 61 text

&NCFE$PSF.-.PEFMJOUP"QQ

Slide 62

Slide 62 text

&NCFE$PSF.-.PEFMJOUP"QQ let model = SRCNN() let highImage = try! model.predict(image: lowImage)

Slide 63

Slide 63 text

.-.PEFM"DDVSBDZ *NQSPWFNFOU

Slide 64

Slide 64 text

"DDFMFSBUJOHUIF4VQFS3FTPMVUJPO $POWPMVUJPOBM/FVSBM/FUXPSL $IBP%POHFUBM

Slide 65

Slide 65 text

'BTU4VQFS3FTPMVUJPOCZ$// '43$//

Slide 66

Slide 66 text

%FNP

Slide 67

Slide 67 text

'43$//QBSBNFUFST ʜ *OQVU -3*NBHF 0VUQVU )3*NBHF ʜ ʜ ʜ ʜ ʜ ʜ < > < > < > < > < > < > < > < > ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 68

Slide 68 text

43$// QBSBNFUFST '43$//QBSBNFUFST ʜ *OQVU -3*NBHF 0VUQVU )3*NBHF ʜ ʜ ʜ ʜ ʜ ʜ < > < > < > < > < > < > < > < > ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 69

Slide 69 text

43$// QBSBNFUFST '43$//QBSBNFUFST ʜ *OQVU -3*NBHF 0VUQVU )3*NBHF ʜ ʜ ʜ ʜ ʜ ʜ < > < > < > < > < > < > < > < > '43$// QBSBNFUFST ϒϥοΫδϟοΫʹΑΖ͘͠ ࠤ౻लๆ Give My Regards to Black Jack SHUHO SATO

Slide 70

Slide 70 text

Performance #JDVCJD 43$// '43$// 1SFEJDUJPO 5JNFQBHF TFD TFD TFD 5SBJOJOH 5JNF IPVST IPVST .-.PEFM 4J[F ,# ,# 14/3 EC EC EC

Slide 71

Slide 71 text

3FDFOUHSFBUQBQFST 1IPUP3FBMJTUJD4JOHMF*NBHF4VQFS3FTPMVUJPO6TJOHB(FOFSBUJWF "EWFSTBSJBM/FUXPSL $ISJTUJBO-FEJHFUBM 3FBM5JNF4JOHMF*NBHFBOE7JEFP4VQFS3FTPMVUJPO6TJOHBO&⒏DJFOU4VC 1JYFM$POWPMVUJPOBM/FVSBM/FUXPSL 8FO[IF4IJFUBM "DDVSBUF*NBHF4VQFS3FTPMVUJPO6TJOH7FSZ%FFQ$POWPMVUJPOBM/FUXPSLT +JXPO,JNFUBM 1JYFM3FDVSTJWF4VQFS3FTPMVUJPO 3ZBO%BIMFUBM

Slide 72

Slide 72 text

This is useful for any type of image. Not only manga.

Slide 73

Slide 73 text

This is useful for any type of image. Not only manga. Give it a try!

Slide 74

Slide 74 text

https://github.com/kenmaz/SuperResolutionKit

Slide 75

Slide 75 text

Open Source https://github.com/kenmaz/SuperResolutionKit

Slide 76

Slide 76 text

Easy to use let imageView: UIImageView = … let image: UIImage = … imageView.setSRImage(image) //Super Resolution⭐ pod “SuperResolutionKit”

Slide 77

Slide 77 text

SuperResolutionKit

Slide 78

Slide 78 text

SuperResolutionKit •UIImageView extension

Slide 79

Slide 79 text

SuperResolutionKit •UIImageView extension •SRConverter (UIImage to UIImage)

Slide 80

Slide 80 text

SuperResolutionKit •UIImageView extension •SRConverter (UIImage to UIImage) •Include pre-trained model (FSRCNN, SRCNN)

Slide 81

Slide 81 text

SuperResolutionKit •UIImageView extension •SRConverter (UIImage to UIImage) •Include pre-trained model (FSRCNN, SRCNN) •Include Python script to train your own model

Slide 82

Slide 82 text

-JNJUBUJPO

Slide 83

Slide 83 text

-JNJUBUJPO w1SFUSBJOFENPEFMQFSGPSNBODF

Slide 84

Slide 84 text

-JNJUBUJPO w1SFUSBJOFENPEFMQFSGPSNBODF w2VBMJUZEFQFOETPOBUZQFTPGJNBHF

Slide 85

Slide 85 text

)PXUPMFWFSBHF.-JO .PCJMF"QQEFWFMPQNFOU

Slide 86

Slide 86 text

.PUJWBUJPOGPS.-JO.PCJMF"QQT

Slide 87

Slide 87 text

.PUJWBUJPOGPS.-JO.PCJMF"QQT 8IZOPUKVTUSVOBMMUIJTPOUIFTFSWFSTJEF #

Slide 88

Slide 88 text

.PUJWBUJPOGPS.-JO.PCJMF"QQT w2VJDLSFTQPOTF 8IZOPUKVTUSVOBMMUIJTPOUIFTFSWFSTJEF #

Slide 89

Slide 89 text

.PUJWBUJPOGPS.-JO.PCJMF"QQT w2VJDLSFTQPOTF w0⒐JOFTVQQPSU 8IZOPUKVTUSVOBMMUIJTPOUIFTFSWFSTJEF #

Slide 90

Slide 90 text

.PUJWBUJPOGPS.-JO.PCJMF"QQT w2VJDLSFTQPOTF w0⒐JOFTVQQPSU w3FEVDJOHTFSWFSBOEOFUXPSLDPTU 8IZOPUKVTUSVOBMMUIJTPOUIFTFSWFSTJEF #

Slide 91

Slide 91 text

.PUJWBUJPOGPS.-JO.PCJMF"QQT w2VJDLSFTQPOTF w0⒐JOFTVQQPSU w3FEVDJOHTFSWFSBOEOFUXPSLDPTU w4FDVSJUZ 8IZOPUKVTUSVOBMMUIJTPOUIFTFSWFSTJEF #

Slide 92

Slide 92 text

$IPPTFUIFCFTUXBZ 5FDIOPMPHZ (PBMT -FWFM 7JTJPOGSBNFXPSL 'BDF5FYU%FUFDUJPO &BTZ $SFBUF.- $VTUPN*NBHF5FYU$MBTTJpDBUJPO &BTZ 5VSJ$SFBUF $VTUPN0CKFDU%FUFDUJPO 4UZMF5SBOTGFS .JE $PSF.- :PVSPXONPEFM 0UIFST $VUUJOHFEHFNFUIPET )BSE

Slide 93

Slide 93 text

4IPVME*MFBSO1ZUIPO

Slide 94

Slide 94 text

4IPVME*MFBSO1ZUIPO w:FT GPSOPX

Slide 95

Slide 95 text

4IPVME*MFBSO1ZUIPO w:FT GPSOPX w1ZUIPO /VN1Z ,FSBT 5FOTPS'MPXʜ

Slide 96

Slide 96 text

4IPVME*MFBSO1ZUIPO w:FT GPSOPX w1ZUIPO /VN1Z ,FSBT 5FOTPS'MPXʜ w1ZUIPOBMUFSOBUJWF

Slide 97

Slide 97 text

4IPVME*MFBSO1ZUIPO w:FT GPSOPX w1ZUIPO /VN1Z ,FSBT 5FOTPS'MPXʜ w1ZUIPOBMUFSOBUJWF w6TF1ZUIPO"1*TXJUI%ZOBNJD.FNCFS-PPLVQ

Slide 98

Slide 98 text

from keras.models import Sequential from keras.layers import Conv2D from keras.optimizers import Adam ch = 1 SRCNN = Sequential(input_shape=(200, 200, ch)) SRCNN.add(Conv2D(128, 9, 9, activation='relu')) SRCNN.add(Conv2D(64, 3, 3, activation='relu')) SRCNN.add(Conv2D(ch, 5, 5, activation='linear')) adam = Adam(lr=0.0003) SRCNN.compile(optimizer=adam, loss='mean_squared_error')

Slide 99

Slide 99 text

let Sequential = Python.import("keras.models.Sequential") let Conv2D = Python.import("keras.layers.Conv2D") let Adam = Python.import("keras.optimizers.Adam") let ch = 1 let SRCNN = Sequential(input_shape: (200, 200, ch)) SRCNN.add(Conv2D(128, (9, 9), activation: 'relu')) SRCNN.add(Conv2D(64, (3, 3), activation: 'relu')) SRCNN.add(Conv2D(ch, (5, 5), activation: 'linear')) let adam = Adam(lr: 0.0003) SRCNN.compile(optimizer: adam, loss: 'mean_squared_error') %ZOBNJD.FNCFS-PPLVQ

Slide 100

Slide 100 text

4IPVME*MFBSO1ZUIPO w:FT GPSOPX w1ZUIPO /VN1Z ,FSBT 5FOTPS'MPXʜ w1ZUIPOBMUFSOBUJWF w6TF1ZUIPO"1*TXJUI%ZOBNJD.FNCFS-PPLVQ

Slide 101

Slide 101 text

4IPVME*MFBSO1ZUIPO w:FT GPSOPX w1ZUIPO /VN1Z ,FSBT 5FOTPS'MPXʜ w1ZUIPOBMUFSOBUJWF w6TF1ZUIPO"1*TXJUI%ZOBNJD.FNCFS-PPLVQ w4XJGUGPS5FOTPS'MPX

Slide 102

Slide 102 text

3FDBQ w*NBHF4VQFS3FTPMVUJPOEFUBJMT w)PXUPJNQMFNFOU*NBHF4VQFS 3FTPMVUJPOXJUI1ZUIPOBOE$PSF.- w4VQFS3FTPMVUJPO,JU 1MFBTFDPOUSJCVUF

Slide 103

Slide 103 text

5IBOL:PV Twitter / github @kenmaz https://github.com/kenmaz/SuperResolutionKit