rename network

README.md

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

src/model.py

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

src/train.py

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