Add Caffe file extension

electron-builder.yml

@@ -28,6 +28,9 @@ fileAssociations:
   - name: "CoreML Model"
     ext:
     - mlmodel
+  - name: "Caffe Model"
+    ext:
+    - caffemodel
 publish:
   - provider: github
     releaseType: release

src/app.js

@@ -108,11 +108,12 @@ class Application {
             properties: [ 'openFile' ], 
             filters: [
                 { name: 'ONNX Model', extensions: [ 'onnx', 'pb' ] },
-                { name: 'TensorFlow Saved Model', extensions: [ 'saved_model.pb' ] },
-                { name: 'TensorFlow Graph', extensions: [ 'pb', 'meta' ] },
-                { name: 'TensorFlow Lite Model', extensions: [ 'tflite' ] },
                 { name: 'Keras Model', extension: [ 'json', 'keras', 'h5' ] },
                 { name: 'CoreML Model', extension: [ 'mlmodel' ] },
+                { name: 'Caffe Model', extension: [ 'caffemodel' ] },
+                { name: 'TensorFlow Graph', extensions: [ 'pb', 'meta' ] },
+                { name: 'TensorFlow Saved Model', extensions: [ 'saved_model.pb' ] },
+                { name: 'TensorFlow Lite Model', extensions: [ 'tflite' ] }
             ]
         };
         electron.dialog.showOpenDialog(showOpenDialogOptions, (selectedFiles) => {