Update README.md

GIMP3-ML
kritiksoman 3 years ago committed by GitHub
parent e3361673aa
commit 9f8a8dcc35
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -65,6 +65,31 @@ cv2.imwrite('output/tmp-matting.png', out) # save as png
out = gimpml.enlighten(image)
cv2.imwrite('output/tmp-enlighten.jpg', out)
face = cv2.imread('sampleinput/face.png')
out = gimpml.parseface(face[:, :, ::-1])
cv2.imwrite('output/tmp-parseface.png', out[:, :, ::-1])
mask1 = cv2.imread('sampleinput/mask1.png')
mask2 = cv2.imread('sampleinput/mask2.png')
out = gimpml.interpolateframe(mask1, mask2, 'output/interpolateframes')
face = cv2.imread('sampleinput/face.png')
out = gimpml.depth(face[:, :, ::-1])
cv2.imwrite('output/tmp-depth.png', out[:, :, ::-1])
image = cv2.imread('sampleinput/face.png')
out = gimpml.semseg(image[:, :, ::-1])
cv2.imwrite('output/tmp-semseg.png', out[:, :, ::-1])
image = cv2.imread('sampleinput/face.png')
out = gimpml.super(image[:, :, ::-1])
cv2.imwrite('output/tmp-super.png', out[:, :, ::-1])
image = cv2.imread('sampleinput/inpaint.png')
mask = cv2.imread('sampleinput/inpaint-mask.png')
out = gimpml.inpaint(image[:, :, ::-1], mask[:, :, 0])
cv2.imwrite('output/tmp-inpaint.png', out[:, :, ::-1])
```
# Model Zoo
@ -82,6 +107,6 @@ cv2.imwrite('output/tmp-enlighten.jpg', out)
| [denoising](https://github.com/kritiksoman/GIMP-ML/wiki/User-Manual#de-noise) | [GPL3](https://github.com/SaoYan/DnCNN-PyTorch/blob/master/LICENSE) | BSD68 |
| [enlighten](https://github.com/kritiksoman/GIMP-ML/wiki/User-Manual#enlightening) | [BSD](https://github.com/VITA-Group/EnlightenGAN/blob/master/License) | [Custom](https://arxiv.org/pdf/1906.06972.pdf) |
| [interpolate-frames](https://github.com/kritiksoman/GIMP-ML/wiki/User-Manual#interpolate-frames) | [MIT](https://github.com/hzwer/arXiv2020-RIFE/blob/main/LICENSE) | [HD](https://arxiv.org/pdf/2011.06294.pdf) |
| [inpainting](https://github.com/kritiksoman/GIMP-ML/wiki/User-Manual#in-painting) | [CC BY-NC-SA 4.0](https://github.com/a-mos/High_Resolution_Image_Inpainting/blob/master/LICENSE.md) | [DIV2K](http://ceur-ws.org/Vol-2744/short18.pdf) |
| [inpainting](https://github.com/kritiksoman/GIMP-ML/wiki/User-Manual#in-painting) | [CC BY-NC 4.0](https://github.com/knazeri/edge-connect/blob/master/LICENSE.md) | [CelebA, CelebHQ, Places2, Paris StreetView](https://openaccess.thecvf.com/content_ICCVW_2019/papers/AIM/Nazeri_EdgeConnect_Structure_Guided_Image_Inpainting_using_Edge_Prediction_ICCVW_2019_paper.pdf) |

Loading…
Cancel
Save