[caffekit] Run inference without caffemodel (#206)

This commit revises 'caffekit run' to performe inference with random weight
even when caffemodel is not provided.

Signed-off-by: Jonghyun Park <jh1302.park@samsung.com>

diff --git a/contrib/caffekit/src/caffekit.cpp b/contrib/caffekit/src/caffekit.cpp
index 3ff0b95..23331c3 100644 (file)
--- a/contrib/caffekit/src/caffekit.cpp
+++ b/contrib/caffekit/src/caffekit.cpp
@@ -17,10 +17,14 @@ public:
 
 int RunCommand::run(int argc, char **argv) const
 {
-  // USAGE: HEADER run [.prototxt] [.caffemodel]
+  // USAGE: HEADER run [.prototxt] ([.caffemodel])
   caffe::Net<float> net(argv[0], caffe::TEST);
 
-  net.CopyTrainedLayersFrom(argv[1]);
+  if (argv[1] != nullptr)
+  {
+    net.CopyTrainedLayersFrom(argv[1]);
+  }
+
   net.Forward();
 
   std::cout << "# of outputs: " << net.num_outputs() << std::endl;