mirror/DeepCreamPy
1
0
Fork 0
mirror of https://github.com/Deepshift/DeepCreamPy.git synced 2025-02-04 15:40:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename network

Browse Source
This commit is contained in:
deep pomf 2018-02-09 20:51:41 -05:00
parent 0071e4585a
commit 5cbafe7a5e
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -16,6 +16,8 @@ Link coming soon
# Usage # Usage
# To do # To do
- Add a user interface - Add a user interface
- Incorporate GAN loss into training - Incorporate GAN loss into training

2
src/network.py → src/model.py
View File

@ -1,6 +1,6 @@
from layer import * from layer import *
class Network: class Model:
def __init__(self, x, mask, local_x, global_completion, local_completion, is_training, batch_size): def __init__(self, x, mask, local_x, global_completion, local_completion, is_training, batch_size):
self.batch_size = batch_size self.batch_size = batch_size
self.imitation = self.generator(x * (1 - mask), is_training) self.imitation = self.generator(x * (1 - mask), is_training)

2
src/train.py
View File

@ -2,7 +2,7 @@ import numpy as np
import tensorflow as tf import tensorflow as tf
import cv2 import cv2
import tqdm import tqdm
from network import Network from model import Model
import load import load
IMAGE_SIZE = 128 IMAGE_SIZE = 128