Added data size and ONNX export options, FP16 inference is working

2 files changed, 6 insertions, 1 deletions

diff --git a/options/base_options.py b/options/base_options.py
index de831fe..561a890 100755
--- a/options/base_options.py
+++ b/options/base_options.py
@@ -56,7 +56,8 @@ class BaseOptions():
         self.parser.add_argument('--n_downsample_E', type=int, default=4, help='# of downsampling layers in encoder') 
         self.parser.add_argument('--nef', type=int, default=16, help='# of encoder filters in the first conv layer')        
         self.parser.add_argument('--n_clusters', type=int, default=10, help='number of clusters for features')
-        
+        self.parser.add_argument('--verbose', action='store_true', default = False, help='toggles verbose')
+
         self.initialized = True
 
     def parse(self, save=True):
diff --git a/options/test_options.py b/options/test_options.py
index aaeff53..504edf3 100755
--- a/options/test_options.py
+++ b/options/test_options.py
@@ -12,4 +12,8 @@ class TestOptions(BaseOptions):
         self.parser.add_argument('--which_epoch', type=str, default='latest', help='which epoch to load? set to latest to use latest cached model')
         self.parser.add_argument('--how_many', type=int, default=50, help='how many test images to run')       
         self.parser.add_argument('--cluster_path', type=str, default='features_clustered_010.npy', help='the path for clustered results of encoded features')
+        self.parser.add_argument("--export_onnx", type=str, help="export ONNX model to a given file")
+        self.parser.add_argument("--engine", type=str, help="run serialized TRT engine")
+        self.parser.add_argument("--onnx", type=str, help="run ONNX model via TRT")
+        self.parser.add_argument("-d", "--data_type", default=32, type=int, choices=[8, 16, 32], help="Supported data type i.e. 8, 16, 32 bit")
         self.isTrain = False