dnn-proto.js 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353
  1. export const dnn = {};
  2. dnn.Model = class Model {
  3. constructor() {
  4. this.input_shape = [];
  5. this.input_name = [];
  6. this.node = [];
  7. this.input = [];
  8. this.output = [];
  9. }
  10. static decode(reader, length) {
  11. const message = new dnn.Model();
  12. const end = length === undefined ? reader.length : reader.position + length;
  13. while (reader.position < end) {
  14. const tag = reader.uint32();
  15. switch (tag >>> 3) {
  16. case 1:
  17. message.name = reader.string();
  18. break;
  19. case 2:
  20. message.version = reader.int32();
  21. break;
  22. case 4:
  23. message.input_shape = reader.array(message.input_shape, () => reader.int32(), tag);
  24. break;
  25. case 7:
  26. message.input_name.push(reader.string());
  27. break;
  28. case 10:
  29. message.node.push(dnn.Node.decode(reader, reader.uint32()));
  30. break;
  31. case 12:
  32. message.input.push(dnn.Parameter.decode(reader, reader.uint32()));
  33. break;
  34. case 13:
  35. message.output.push(dnn.Parameter.decode(reader, reader.uint32()));
  36. break;
  37. case 14:
  38. message.a014 = reader.double();
  39. break;
  40. default:
  41. reader.skipType(tag & 7);
  42. break;
  43. }
  44. }
  45. return message;
  46. }
  47. };
  48. dnn.Model.prototype.name = "";
  49. dnn.Model.prototype.version = 0;
  50. dnn.Model.prototype.a014 = 0;
  51. dnn.Parameter = class Parameter {
  52. static decode(reader, length) {
  53. const message = new dnn.Parameter();
  54. const end = length === undefined ? reader.length : reader.position + length;
  55. while (reader.position < end) {
  56. const tag = reader.uint32();
  57. switch (tag >>> 3) {
  58. case 1:
  59. message.name = reader.string();
  60. break;
  61. case 2:
  62. message.shape = dnn.Shape.decode(reader, reader.uint32());
  63. break;
  64. default:
  65. reader.skipType(tag & 7);
  66. break;
  67. }
  68. }
  69. return message;
  70. }
  71. };
  72. dnn.Parameter.prototype.name = "";
  73. dnn.Parameter.prototype.shape = null;
  74. dnn.Shape = class Shape {
  75. static decode(reader, length) {
  76. const message = new dnn.Shape();
  77. const end = length === undefined ? reader.length : reader.position + length;
  78. while (reader.position < end) {
  79. const tag = reader.uint32();
  80. switch (tag >>> 3) {
  81. case 1:
  82. message.dim0 = reader.int32();
  83. break;
  84. case 2:
  85. message.dim1 = reader.int32();
  86. break;
  87. case 3:
  88. message.dim2 = reader.int32();
  89. break;
  90. case 4:
  91. message.dim3 = reader.int32();
  92. break;
  93. default:
  94. reader.skipType(tag & 7);
  95. break;
  96. }
  97. }
  98. return message;
  99. }
  100. };
  101. dnn.Shape.prototype.dim0 = 0;
  102. dnn.Shape.prototype.dim1 = 0;
  103. dnn.Shape.prototype.dim2 = 0;
  104. dnn.Shape.prototype.dim3 = 0;
  105. dnn.Node = class Node {
  106. constructor() {
  107. this.input = [];
  108. this.output = [];
  109. }
  110. static decode(reader, length) {
  111. const message = new dnn.Node();
  112. const end = length === undefined ? reader.length : reader.position + length;
  113. while (reader.position < end) {
  114. const tag = reader.uint32();
  115. switch (tag >>> 3) {
  116. case 1:
  117. message.layer = dnn.Layer.decode(reader, reader.uint32());
  118. break;
  119. case 2:
  120. message.input.push(reader.string());
  121. break;
  122. case 3:
  123. message.output.push(reader.string());
  124. break;
  125. default:
  126. reader.skipType(tag & 7);
  127. break;
  128. }
  129. }
  130. return message;
  131. }
  132. };
  133. dnn.Node.prototype.layer = null;
  134. dnn.Layer = class Layer {
  135. constructor() {
  136. this.weight = [];
  137. }
  138. static decode(reader, length) {
  139. const message = new dnn.Layer();
  140. const end = length === undefined ? reader.length : reader.position + length;
  141. while (reader.position < end) {
  142. const tag = reader.uint32();
  143. switch (tag >>> 3) {
  144. case 1:
  145. message.name = reader.string();
  146. break;
  147. case 2:
  148. message.type = reader.string();
  149. break;
  150. case 3:
  151. message.filters = reader.int32();
  152. break;
  153. case 7:
  154. message.a007 = reader.int32();
  155. break;
  156. case 8:
  157. message.a008 = reader.int32();
  158. break;
  159. case 9:
  160. message.groups = reader.int32();
  161. break;
  162. case 10:
  163. message.a010 = reader.int32();
  164. break;
  165. case 11:
  166. message.a011 = reader.int32();
  167. break;
  168. case 14:
  169. message.slope = reader.float();
  170. break;
  171. case 15:
  172. message.intercept = reader.float();
  173. break;
  174. case 50:
  175. message.weight.push(dnn.Tensor.decode(reader, reader.uint32()));
  176. break;
  177. case 72:
  178. message.operation = reader.int32();
  179. break;
  180. case 65:
  181. message.axis = reader.int32();
  182. break;
  183. case 77:
  184. message.a077 = reader.int32();
  185. break;
  186. case 79:
  187. message.scale = reader.float();
  188. break;
  189. case 80:
  190. message.pad_1 = reader.int32();
  191. break;
  192. case 81:
  193. message.pad_2 = reader.int32();
  194. break;
  195. case 82:
  196. message.pad_3 = reader.int32();
  197. break;
  198. case 83:
  199. message.pad_4 = reader.int32();
  200. break;
  201. case 84:
  202. message.pad_5 = reader.int32();
  203. break;
  204. case 85:
  205. message.a085 = reader.int32();
  206. break;
  207. case 90:
  208. message.a090 = reader.int32();
  209. break;
  210. case 101:
  211. message.is_quantized = reader.bool();
  212. break;
  213. case 104:
  214. message.quantization = dnn.Buffer.decode(reader, reader.uint32());
  215. break;
  216. case 109:
  217. message.stride_w = reader.int32();
  218. break;
  219. case 110:
  220. message.stride_h = reader.int32();
  221. break;
  222. case 111:
  223. message.kernel_w = reader.int32();
  224. break;
  225. case 112:
  226. message.kernel_h = reader.int32();
  227. break;
  228. case 115:
  229. message.a115 = reader.int32();
  230. break;
  231. case 116:
  232. message.a116 = reader.int32();
  233. break;
  234. default:
  235. reader.skipType(tag & 7);
  236. break;
  237. }
  238. }
  239. return message;
  240. }
  241. };
  242. dnn.Layer.prototype.name = "";
  243. dnn.Layer.prototype.type = "";
  244. dnn.Layer.prototype.filters = 0;
  245. dnn.Layer.prototype.a007 = 0;
  246. dnn.Layer.prototype.a008 = 0;
  247. dnn.Layer.prototype.groups = 0;
  248. dnn.Layer.prototype.a010 = 0;
  249. dnn.Layer.prototype.a011 = 0;
  250. dnn.Layer.prototype.slope = 0;
  251. dnn.Layer.prototype.intercept = 0;
  252. dnn.Layer.prototype.operation = 0;
  253. dnn.Layer.prototype.axis = 0;
  254. dnn.Layer.prototype.a077 = 0;
  255. dnn.Layer.prototype.scale = 0;
  256. dnn.Layer.prototype.pad_1 = 0;
  257. dnn.Layer.prototype.pad_2 = 0;
  258. dnn.Layer.prototype.pad_3 = 0;
  259. dnn.Layer.prototype.pad_4 = 0;
  260. dnn.Layer.prototype.pad_5 = 0;
  261. dnn.Layer.prototype.a085 = 0;
  262. dnn.Layer.prototype.a090 = 0;
  263. dnn.Layer.prototype.is_quantized = false;
  264. dnn.Layer.prototype.quantization = null;
  265. dnn.Layer.prototype.stride_w = 0;
  266. dnn.Layer.prototype.stride_h = 0;
  267. dnn.Layer.prototype.kernel_w = 0;
  268. dnn.Layer.prototype.kernel_h = 0;
  269. dnn.Layer.prototype.a115 = 0;
  270. dnn.Layer.prototype.a116 = 0;
  271. dnn.Buffer = class Buffer {
  272. static decode(reader, length) {
  273. const message = new dnn.Buffer();
  274. const end = length === undefined ? reader.length : reader.position + length;
  275. while (reader.position < end) {
  276. const tag = reader.uint32();
  277. switch (tag >>> 3) {
  278. case 5:
  279. message.data = reader.bytes();
  280. break;
  281. default:
  282. reader.skipType(tag & 7);
  283. break;
  284. }
  285. }
  286. return message;
  287. }
  288. };
  289. dnn.Buffer.prototype.data = new Uint8Array([]);
  290. dnn.Tensor = class Tensor {
  291. static decode(reader, length) {
  292. const message = new dnn.Tensor();
  293. const end = length === undefined ? reader.length : reader.position + length;
  294. while (reader.position < end) {
  295. const tag = reader.uint32();
  296. switch (tag >>> 3) {
  297. case 1:
  298. message.dim0 = reader.int32();
  299. break;
  300. case 2:
  301. message.dim1 = reader.int32();
  302. break;
  303. case 3:
  304. message.dim2 = reader.int32();
  305. break;
  306. case 4:
  307. message.dim3 = reader.int32();
  308. break;
  309. case 5:
  310. message.data = reader.bytes();
  311. break;
  312. case 6:
  313. message.quantized_data = reader.bytes();
  314. break;
  315. default:
  316. reader.skipType(tag & 7);
  317. break;
  318. }
  319. }
  320. return message;
  321. }
  322. };
  323. dnn.Tensor.prototype.dim0 = 0;
  324. dnn.Tensor.prototype.dim1 = 0;
  325. dnn.Tensor.prototype.dim2 = 0;
  326. dnn.Tensor.prototype.dim3 = 0;
  327. dnn.Tensor.prototype.data = new Uint8Array([]);
  328. dnn.Tensor.prototype.quantized_data = new Uint8Array([]);