pic

.gitignore

@@ -108,6 +108,7 @@ venv.bak/
 
 data/
 pretrained/
+paper/
 
 *.DS_Store
 

README.md

@@ -28,6 +28,9 @@ The predicted samples can be seen in `./samples` folder. The detailed arguments
 
 The test predicted samples can be seen in `./samples/test` folder
 
+## Framework
+![framework](https://www.dropbox.com/s/rbaqbe2gy1pdzyq/framework0.pdf?raw=1)
+
 ## Cite
 If you use this code or reference our paper in your work please cite this publication as:
 ```

src/seenet.py

@@ -355,7 +355,7 @@ def train(self, loader, test_loader):
         prefix = time.strftime("%Y_%d_%m_%H_%M_%S", time.localtime())
         summary_dir = self.log_dir + prefix + '/'
         writer = tf.summary.FileWriter(summary_dir, self.sess.graph)
-        params = ("KTH_MCNET"
+        params = ("SEENET"
                   + "_lr_c=" + str(self.lr_c)
                   + "_lr_m=" + str(self.lr_m)
                   + "_margin=" + str(self.margin)