Caffe2 .pbtxt detection (#223)

src/caffe2.js

@@ -43,8 +43,10 @@ caffe2.ModelFactory = class {
                 }
             }
         }
-        if (identifier.endsWith('predict_net.pbtxt') || identifier.endsWith('predict_net.prototxt') ||
-            identifier.endsWith('init_net.pbtxt') || identifier.endsWith('init_net.prototxt')) {
+        if (extension == 'pbtxt' || extension == 'prototxt') {
+            if (identifier.endsWith('predict_net.pbtxt') || identifier.endsWith('predict_net.prototxt')) {
+                return true;
+            }
             tags = context.tags('pbtxt');
             if (tags.op) {
                 return true;

src/view.js

@@ -369,7 +369,7 @@ view.View = class {
         setTimeout(() => {
             if (graph && graph != this._activeGraph) {
                 var nodes = graph.nodes;
-                if (nodes.length > 1500) {
+                if (nodes.length > 1400) {
                     if (!this._host.confirm('Large model detected.', 'This graph contains a large number of nodes and might take a long time to render. Do you want to continue?')) {
                         this._host.event('Graph', 'Render', 'Skip', nodes.length);
                         this.show(null);

test/models.json

@@ -667,7 +667,6 @@
     "format": "Caffe2",
     "link":   "https://github.com/jolibrain/deepdetect/tree/master/templates/caffe2"
   },
-
   {
     "type":   "caffe2",
     "target": "bvlc_alexnet/predict_net.pb,bvlc_alexnet/init_net.pb",
@@ -731,6 +730,14 @@
     "format": "Caffe2",
     "link":   "https://github.com/zyddawn/caffe2-models"
   },
+  {
+    "type":   "caffe2",
+    "target": "generalized_rcnn/net.pbtxt",
+    "source": "https://github.com/lutzroeder/netron/files/2801714/net.pbtxt.txt",
+    "format": "Caffe2",
+    "render": "skip",
+    "link":   "https://github.com/lutzroeder/netron/issues/223"
+  },
   {
     "type":   "caffe2",
     "target": "inception_v1/predict_net.pb,inception_v1/init_net.pb",