mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2025-01-05 21:55:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

update readme

Browse Source
This commit is contained in:
deeppomf 2018-02-21 22:38:08 -05:00
parent d6348a0d5d
commit fe86070b74
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
README.md
View File

@ -5,6 +5,8 @@ This project applies an implementation of [Globally and Locally Consistent Image
# **THIS PROJECT IS STILL IN DEVELOPMENT. DO NOT BE DISAPPOINTED IF THE RESULTS AREN'T AS GOOD AS YOU EXPECT.**
Next planned update on March 1: Incorporate GAN loss in training. Improve CLI.
![Censored, decensored](/readme_images/collage.png)
# Limitations
@ -81,9 +83,11 @@ Train pretrained model on your custom dataset.
$ python train.py
```
Training can be done separately for mosaics with train_mosaic.py, but decensor.py is not yet compatible with mosaic decensorship models.
# To do
- ~~Add Python 3 compatibility~~
- Add random rotations in cropping rectangles
- ~~Add random rotations in cropping rectangles~~
- Retrain for arbitrary shape censors
- Add a user interface
- Incorporate GAN loss into training