From f856f6e7f1fd0a11244527b266c1fd47334fb5f3 Mon Sep 17 00:00:00 2001
From: Edmond Yoo <hj3yoo@uwaterloo.ca>
Date: Sun, 16 Sep 2018 01:54:09 +0000
Subject: [PATCH] 10-class model for card detection

---
 src/parser.h |    6 ++++++
 1 files changed, 6 insertions(+), 0 deletions(-)

diff --git a/src/parser.h b/src/parser.h
index 891e658..b760116 100644
--- a/src/parser.h
+++ b/src/parser.h
@@ -3,6 +3,12 @@
 #include "network.h"
 
 network parse_network_cfg(char *filename);
+network parse_network_cfg_custom(char *filename, int batch);
 void save_network(network net, char *filename);
+void save_weights(network net, char *filename);
+void save_weights_upto(network net, char *filename, int cutoff);
+void save_weights_double(network net, char *filename);
+void load_weights(network *net, char *filename);
+void load_weights_upto(network *net, char *filename, int cutoff);
 
 #endif

--
Gitblit v1.10.0