caffe-metadata.json 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525
  1. [
  2. {
  3. "name": "Convolution",
  4. "schema": {
  5. "category": "Layer",
  6. "inputs": [
  7. { "name": "input" },
  8. { "name": "filter" },
  9. { "name": "bias" }
  10. ],
  11. "outputs": [
  12. { "name": "output" }
  13. ],
  14. "attributes": [
  15. { "name": "bias_term", "visible": false },
  16. { "name": "weight_filler", "visible": false },
  17. { "name": "bias_filler", "visible": false },
  18. { "name": "num_output", "visible": false },
  19. { "name": "pad", "default": [0] },
  20. { "name": "kernel_size", "default": [] },
  21. { "name": "stride", "default": [1] },
  22. { "name": "dilation", "default": [] },
  23. { "name": "group", "default": 1 }
  24. ]
  25. }
  26. },
  27. {
  28. "name": "Deconvolution",
  29. "schema": {
  30. "category": "Layer",
  31. "inputs": [
  32. { "name": "input" },
  33. { "name": "filter" },
  34. { "name": "bias" }
  35. ],
  36. "outputs": [
  37. { "name": "output" }
  38. ],
  39. "attributes": [
  40. { "name": "bias_term", "visible": false },
  41. { "name": "weight_filler", "visible": false },
  42. { "name": "bias_filler", "visible": false },
  43. { "name": "num_output", "visible": false },
  44. { "name": "pad", "default": [] },
  45. { "name": "kernel_size", "default": [] },
  46. { "name": "stride", "default": [] },
  47. { "name": "dilation", "default": [] }
  48. ]
  49. }
  50. },
  51. {
  52. "name": "DepthwiseConvolution",
  53. "schema": {
  54. "category": "Layer",
  55. "attributes": [
  56. { "name": "bias_term", "visible": false },
  57. { "name": "weight_filler", "visible": false },
  58. { "name": "bias_filler", "visible": false },
  59. { "name": "num_output", "visible": false }
  60. ],
  61. "inputs": [
  62. { "name": "input" },
  63. { "name": "filter" },
  64. { "name": "bias" }
  65. ],
  66. "outputs": [
  67. { "name": "output" }
  68. ]
  69. }
  70. },
  71. {
  72. "name": "ConvolutionDepthwise",
  73. "schema": {
  74. "category": "Layer",
  75. "attributes": [
  76. { "name": "pad", "default": [0] },
  77. { "name": "kernel_size", "default": [] },
  78. { "name": "stride", "default": [1] },
  79. { "name": "bias_term", "visible": false },
  80. { "name": "weight_filler", "visible": false },
  81. { "name": "bias_filler", "visible": false },
  82. { "name": "num_output", "visible": false }
  83. ],
  84. "inputs": [
  85. { "name": "input" },
  86. { "name": "filter" },
  87. { "name": "bias" }
  88. ],
  89. "outputs": [
  90. { "name": "output" }
  91. ]
  92. }
  93. },
  94. {
  95. "name": "InnerProduct",
  96. "schema": {
  97. "category": "Layer",
  98. "inputs": [
  99. { "name": "input" },
  100. { "name": "weights" },
  101. { "name": "bias" }
  102. ],
  103. "outputs": [
  104. { "name": "output" }
  105. ],
  106. "attributes": [
  107. { "name": "bias_term", "visible": false },
  108. { "name": "weight_filler", "visible": false },
  109. { "name": "bias_filler", "visible": false },
  110. { "name": "num_output", "visible": false }
  111. ]
  112. }
  113. },
  114. {
  115. "name": "Scale",
  116. "schema": {
  117. "category": "Layer",
  118. "inputs": [
  119. { "name": "input" },
  120. { "name": "scale" },
  121. { "name": "bias" }
  122. ],
  123. "outputs": [
  124. { "name": "output" }
  125. ],
  126. "attributes": [
  127. { "name": "filler", "visible": false },
  128. { "name": "bias_term", "visible": false },
  129. { "name": "bias_filler", "visible": false }
  130. ]
  131. }
  132. },
  133. {
  134. "name": "Dropout",
  135. "schema": {
  136. "category": "Dropout",
  137. "attributes": [
  138. { "name": "dropout_ratio", "default": 0.5 }
  139. ],
  140. "inputs": [
  141. { "name": "input" }
  142. ],
  143. "outputs": [
  144. { "name": "output" }
  145. ]
  146. }
  147. },
  148. {
  149. "name": "Flatten",
  150. "schema": {
  151. "category": "Shape",
  152. "inputs": [
  153. { "name": "input" }
  154. ],
  155. "outputs": [
  156. { "name": "output" }
  157. ]
  158. }
  159. },
  160. {
  161. "name": "LRN",
  162. "schema": {
  163. "category": "Normalization",
  164. "attributes": [
  165. { "name": "local_size", "type": "uint32", "default": 5 },
  166. { "name": "alpha", "type": "float32", "default": 0.0001 },
  167. { "name": "beta", "type": "float32", "default": 0.75 }
  168. ],
  169. "inputs": [
  170. { "name": "input" }
  171. ],
  172. "outputs": [
  173. { "name": "output" }
  174. ]
  175. }
  176. },
  177. {
  178. "name": "BatchNorm",
  179. "schema": {
  180. "category": "Normalization",
  181. "attributes": [
  182. { "name": "use_global_stats", "visible": false },
  183. { "name": "eps", "default": 1e-5 }
  184. ],
  185. "inputs": [
  186. { "name": "input" },
  187. { "name": "gamma" },
  188. { "name": "beta" },
  189. { "name": "mean" },
  190. { "name": "variance" }
  191. ],
  192. "outputs": [
  193. { "name": "output" }
  194. ]
  195. }
  196. },
  197. {
  198. "name": "BN",
  199. "schema": {
  200. "category": "Normalization",
  201. "inputs": [
  202. { "name": "input" }
  203. ],
  204. "outputs": [
  205. { "name": "output" }
  206. ]
  207. }
  208. },
  209. {
  210. "name": "Sigmoid",
  211. "schema": {
  212. "category": "Activation",
  213. "inputs": [
  214. { "name": "input" }
  215. ],
  216. "outputs": [
  217. { "name": "output" }
  218. ]
  219. }
  220. },
  221. {
  222. "name": "Softmax",
  223. "schema": {
  224. "category": "Activation",
  225. "inputs": [
  226. { "name": "input" }
  227. ],
  228. "outputs": [
  229. { "name": "output" }
  230. ]
  231. }
  232. },
  233. {
  234. "name": "SoftmaxLoss",
  235. "schema": {
  236. "category": "Activation",
  237. "inputs": [
  238. { "name": "input" },
  239. { "name": "labels" }
  240. ],
  241. "outputs": [
  242. { "name": "output" }
  243. ]
  244. }
  245. },
  246. {
  247. "name": "SoftmaxWithLoss",
  248. "schema": {
  249. "category": "Activation",
  250. "inputs": [
  251. { "name": "input" },
  252. { "name": "labels" }
  253. ],
  254. "outputs": [
  255. { "name": "output" }
  256. ]
  257. }
  258. },
  259. {
  260. "name": "ContrastiveLossParameter",
  261. "schema": {
  262. "attributes": [
  263. { "name": "margin", "default": 1.0 },
  264. { "name": "legacy_version", "default": false }
  265. ]
  266. }
  267. },
  268. {
  269. "name": "ReLU",
  270. "schema": {
  271. "category": "Activation",
  272. "inputs": [
  273. { "name": "input" }
  274. ],
  275. "outputs": [
  276. { "name": "output" }
  277. ]
  278. }
  279. },
  280. {
  281. "name": "PReLU",
  282. "schema": {
  283. "category": "Activation",
  284. "inputs": [
  285. { "name": "input" },
  286. { "name": "slope" }
  287. ],
  288. "outputs": [
  289. { "name": "output" }
  290. ]
  291. }
  292. },
  293. {
  294. "name": "Concat",
  295. "schema": {
  296. "category": "Tensor",
  297. "inputs": [
  298. { "name": "inputs", "option": "variadic" }
  299. ],
  300. "outputs": [
  301. { "name": "output" }
  302. ]
  303. }
  304. },
  305. {
  306. "name": "Split",
  307. "schema": {
  308. "category": "Tensor",
  309. "inputs": [
  310. { "name": "input" }
  311. ],
  312. "outputs": [
  313. { "name": "outputs", "option": "variadic" }
  314. ]
  315. }
  316. },
  317. {
  318. "name": "Eltwise",
  319. "schema": {
  320. "attributes": [
  321. { "name": "operation", "default": 1 }
  322. ],
  323. "inputs": [
  324. { "name": "inputs", "option": "variadic" }
  325. ],
  326. "outputs": [
  327. { "name": "output" }
  328. ]
  329. }
  330. },
  331. {
  332. "name": "Pooling",
  333. "schema": {
  334. "category": "Pool",
  335. "attributes": [
  336. { "name": "pool", "default": 0 }
  337. ],
  338. "inputs": [
  339. { "name": "input" }
  340. ],
  341. "outputs": [
  342. { "name": "output" }
  343. ]
  344. }
  345. },
  346. {
  347. "name": "Crop",
  348. "schema": {
  349. "category": "Data",
  350. "inputs": [
  351. { "name": "data" },
  352. { "name": "size" }
  353. ],
  354. "outputs": [
  355. { "name": "output" }
  356. ]
  357. }
  358. },
  359. {
  360. "name": "Data",
  361. "schema": {
  362. "category": "Data",
  363. "outputs": [
  364. { "name": "data" },
  365. { "name": "label" }
  366. ]
  367. }
  368. },
  369. {
  370. "name": "DummyData",
  371. "schema": {
  372. "category": "Data",
  373. "outputs": [
  374. { "name": "data" }
  375. ]
  376. }
  377. },
  378. {
  379. "name": "AnnotatedData",
  380. "schema": {
  381. "category": "Data",
  382. "outputs": [
  383. { "name": "data" }
  384. ]
  385. }
  386. },
  387. {
  388. "name": "HDF5Data",
  389. "schema": {
  390. "category": "Data",
  391. "outputs": [
  392. { "name": "data" }
  393. ]
  394. }
  395. },
  396. {
  397. "name": "ImageData",
  398. "schema": {
  399. "category": "Data",
  400. "outputs": [
  401. { "name": "data" },
  402. { "name": "label" }
  403. ]
  404. }
  405. },
  406. {
  407. "name": "WindowData",
  408. "schema": {
  409. "category": "Data",
  410. "outputs": [
  411. { "name": "data" },
  412. { "name": "label" }
  413. ]
  414. }
  415. },
  416. {
  417. "name": "Slice",
  418. "schema": {
  419. "category": "Tensor",
  420. "attributes": [
  421. { "name": "axis", "default": 1 }
  422. ],
  423. "inputs": [
  424. { "name": "input" }
  425. ],
  426. "outputs": [
  427. { "name": "outputs", "option": "variadic" }
  428. ]
  429. }
  430. },
  431. {
  432. "name": "EuclideanLoss",
  433. "schema": {
  434. "inputs": [
  435. { "name": "predictions" },
  436. { "name": "targets" }
  437. ],
  438. "outputs": [
  439. { "name": "output" }
  440. ]
  441. }
  442. },
  443. {
  444. "name": "Accuracy",
  445. "schema": {
  446. "inputs": [
  447. { "name": "predictions" },
  448. { "name": "labels" }
  449. ],
  450. "outputs": [
  451. { "name": "output" }
  452. ]
  453. }
  454. },
  455. {
  456. "name": "LSTM",
  457. "schema": {
  458. "category": "Layer",
  459. "inputs": [
  460. { "name": "input" },
  461. { "name": "weights" },
  462. { "name": "h_0" },
  463. { "name": "c_0" }
  464. ],
  465. "outputs": [
  466. { "name": "output" },
  467. { "name": "h_T" },
  468. { "name": "c_T" }
  469. ],
  470. "attributes": [
  471. { "name": "weight_filler", "visible": false },
  472. { "name": "bias_filler", "visible": false },
  473. { "name": "num_output", "visible": false }
  474. ]
  475. }
  476. },
  477. {
  478. "name": "Reshape",
  479. "schema": {
  480. "category": "Shape",
  481. "inputs": [
  482. { "name": "data" }
  483. ],
  484. "outputs": [
  485. { "name": "reshaped" }
  486. ]
  487. }
  488. },
  489. {
  490. "name": "ColorConv",
  491. "schema": {
  492. "inputs": [
  493. { "name": "input" }
  494. ],
  495. "outputs": [
  496. { "name": "output" }
  497. ]
  498. }
  499. },
  500. {
  501. "name": "Permute",
  502. "schema": {
  503. "category": "Shape",
  504. "inputs": [
  505. { "name": "input" }
  506. ],
  507. "outputs": [
  508. { "name": "output" }
  509. ]
  510. }
  511. },
  512. {
  513. "name": "Parameter",
  514. "schema": {
  515. "outputs": [
  516. { "name": "output" }
  517. ]
  518. }
  519. },
  520. {
  521. "name": "Python",
  522. "schema": {
  523. }
  524. }
  525. ]