SqueezeNet_deploy.prototxt 15 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873
  1. # SqueezeNet architecture for image classification on ImageNet dataset
  2. name: "SqueezeNet"
  3. input: "data"
  4. input_dim: 1
  5. input_dim: 3
  6. input_dim: 416
  7. input_dim: 416
  8. layer {
  9. name: "conv1"
  10. type: "Convolution"
  11. bottom: "data"
  12. top: "conv1"
  13. convolution_param {
  14. num_output: 96
  15. kernel_size: 7
  16. stride: 2
  17. weight_filler {
  18. type: "xavier"
  19. }
  20. bias_filler {
  21. type: "constant"
  22. value: 0.01
  23. }
  24. }
  25. }
  26. layer {
  27. name: "rect_conv1"
  28. type: "ReLU"
  29. relu_param {
  30. negative_slope: 0.01
  31. }
  32. bottom: "conv1"
  33. top: "conv1"
  34. }
  35. layer {
  36. name: "pool1"
  37. type: "Pooling"
  38. bottom: "conv1"
  39. top: "pool1"
  40. pooling_param {
  41. pool: MAX
  42. kernel_size: 3
  43. stride: 2
  44. }
  45. }
  46. layer {
  47. name: "fire2_squeeze"
  48. type: "Convolution"
  49. bottom: "pool1"
  50. top: "fire2_squeeze"
  51. convolution_param {
  52. num_output: 16
  53. kernel_size: 1
  54. stride: 1
  55. pad: 0
  56. weight_filler {
  57. type: "xavier"
  58. }
  59. bias_filler {
  60. type: "constant"
  61. value: 0.01
  62. }
  63. }
  64. }
  65. layer {
  66. name: "rect_fire2_squeeze"
  67. type: "ReLU"
  68. relu_param {
  69. negative_slope: 0.01
  70. }
  71. bottom: "fire2_squeeze"
  72. top: "fire2_squeeze"
  73. }
  74. layer {
  75. name: "fire2_expand_1x1"
  76. type: "Convolution"
  77. bottom: "fire2_squeeze"
  78. top: "fire2_expand_1x1"
  79. convolution_param {
  80. num_output: 64
  81. kernel_size: 1
  82. stride: 1
  83. pad: 0
  84. weight_filler {
  85. type: "xavier"
  86. }
  87. bias_filler {
  88. type: "constant"
  89. value: 0.01
  90. }
  91. }
  92. }
  93. layer {
  94. name: "rect_fire2_expand_1x1"
  95. type: "ReLU"
  96. relu_param {
  97. negative_slope: 0.01
  98. }
  99. bottom: "fire2_expand_1x1"
  100. top: "fire2_expand_1x1"
  101. }
  102. layer {
  103. name: "fire2_expand_3x3"
  104. type: "Convolution"
  105. bottom: "fire2_squeeze"
  106. top: "fire2_expand_3x3"
  107. convolution_param {
  108. num_output: 64
  109. kernel_size: 3
  110. stride: 1
  111. pad: 1
  112. weight_filler {
  113. type: "xavier"
  114. }
  115. bias_filler {
  116. type: "constant"
  117. value: 0.01
  118. }
  119. }
  120. }
  121. layer {
  122. name: "rect_fire2_expand_3x3"
  123. type: "ReLU"
  124. relu_param {
  125. negative_slope: 0.01
  126. }
  127. bottom: "fire2_expand_3x3"
  128. top: "fire2_expand_3x3"
  129. }
  130. layer {
  131. name: "fire2"
  132. type: "Concat"
  133. bottom: "fire2_expand_1x1"
  134. bottom: "fire2_expand_3x3"
  135. top: "fire2"
  136. concat_param {
  137. axis: 1
  138. }
  139. }
  140. layer {
  141. name: "fire3_squeeze"
  142. type: "Convolution"
  143. bottom: "fire2"
  144. top: "fire3_squeeze"
  145. convolution_param {
  146. num_output: 16
  147. kernel_size: 1
  148. stride: 1
  149. pad: 0
  150. weight_filler {
  151. type: "xavier"
  152. }
  153. bias_filler {
  154. type: "constant"
  155. value: 0.01
  156. }
  157. }
  158. }
  159. layer {
  160. name: "rect_fire3_squeeze"
  161. type: "ReLU"
  162. relu_param {
  163. negative_slope: 0.01
  164. }
  165. bottom: "fire3_squeeze"
  166. top: "fire3_squeeze"
  167. }
  168. layer {
  169. name: "fire3_expand_1x1"
  170. type: "Convolution"
  171. bottom: "fire3_squeeze"
  172. top: "fire3_expand_1x1"
  173. convolution_param {
  174. num_output: 64
  175. kernel_size: 1
  176. stride: 1
  177. pad: 0
  178. weight_filler {
  179. type: "xavier"
  180. }
  181. bias_filler {
  182. type: "constant"
  183. value: 0.01
  184. }
  185. }
  186. }
  187. layer {
  188. name: "rect_fire3_expand_1x1"
  189. type: "ReLU"
  190. relu_param {
  191. negative_slope: 0.01
  192. }
  193. bottom: "fire3_expand_1x1"
  194. top: "fire3_expand_1x1"
  195. }
  196. layer {
  197. name: "fire3_expand_3x3"
  198. type: "Convolution"
  199. bottom: "fire3_squeeze"
  200. top: "fire3_expand_3x3"
  201. convolution_param {
  202. num_output: 64
  203. kernel_size: 3
  204. stride: 1
  205. pad: 1
  206. weight_filler {
  207. type: "xavier"
  208. }
  209. bias_filler {
  210. type: "constant"
  211. value: 0.01
  212. }
  213. }
  214. }
  215. layer {
  216. name: "rect_fire3_expand_3x3"
  217. type: "ReLU"
  218. relu_param {
  219. negative_slope: 0.01
  220. }
  221. bottom: "fire3_expand_3x3"
  222. top: "fire3_expand_3x3"
  223. }
  224. layer {
  225. name: "fire3"
  226. type: "Concat"
  227. bottom: "fire3_expand_1x1"
  228. bottom: "fire3_expand_3x3"
  229. top: "fire3"
  230. concat_param {
  231. axis: 1
  232. }
  233. }
  234. layer {
  235. name: "fire4_squeeze"
  236. type: "Convolution"
  237. bottom: "fire3"
  238. top: "fire4_squeeze"
  239. convolution_param {
  240. num_output: 32
  241. kernel_size: 1
  242. stride: 1
  243. pad: 0
  244. weight_filler {
  245. type: "xavier"
  246. }
  247. bias_filler {
  248. type: "constant"
  249. value: 0.01
  250. }
  251. }
  252. }
  253. layer {
  254. name: "rect_fire4_squeeze"
  255. type: "ReLU"
  256. relu_param {
  257. negative_slope: 0.01
  258. }
  259. bottom: "fire4_squeeze"
  260. top: "fire4_squeeze"
  261. }
  262. layer {
  263. name: "fire4_expand_1x1"
  264. type: "Convolution"
  265. bottom: "fire4_squeeze"
  266. top: "fire4_expand_1x1"
  267. convolution_param {
  268. num_output: 128
  269. kernel_size: 1
  270. stride: 1
  271. pad: 0
  272. weight_filler {
  273. type: "xavier"
  274. }
  275. bias_filler {
  276. type: "constant"
  277. value: 0.01
  278. }
  279. }
  280. }
  281. layer {
  282. name: "rect_fire4_expand_1x1"
  283. type: "ReLU"
  284. relu_param {
  285. negative_slope: 0.01
  286. }
  287. bottom: "fire4_expand_1x1"
  288. top: "fire4_expand_1x1"
  289. }
  290. layer {
  291. name: "fire4_expand_3x3"
  292. type: "Convolution"
  293. bottom: "fire4_squeeze"
  294. top: "fire4_expand_3x3"
  295. convolution_param {
  296. num_output: 128
  297. kernel_size: 3
  298. stride: 1
  299. pad: 1
  300. weight_filler {
  301. type: "xavier"
  302. }
  303. bias_filler {
  304. type: "constant"
  305. value: 0.01
  306. }
  307. }
  308. }
  309. layer {
  310. name: "rect_fire4_expand_3x3"
  311. type: "ReLU"
  312. relu_param {
  313. negative_slope: 0.01
  314. }
  315. bottom: "fire4_expand_3x3"
  316. top: "fire4_expand_3x3"
  317. }
  318. layer {
  319. name: "fire4"
  320. type: "Concat"
  321. bottom: "fire4_expand_1x1"
  322. bottom: "fire4_expand_3x3"
  323. top: "fire4"
  324. concat_param {
  325. axis: 1
  326. }
  327. }
  328. layer {
  329. name: "pool4"
  330. type: "Pooling"
  331. bottom: "fire4"
  332. top: "pool4"
  333. pooling_param {
  334. pool: MAX
  335. kernel_size: 3
  336. stride: 2
  337. }
  338. }
  339. layer {
  340. name: "fire5_squeeze"
  341. type: "Convolution"
  342. bottom: "pool4"
  343. top: "fire5_squeeze"
  344. convolution_param {
  345. num_output: 32
  346. kernel_size: 1
  347. stride: 1
  348. pad: 0
  349. weight_filler {
  350. type: "xavier"
  351. }
  352. bias_filler {
  353. type: "constant"
  354. value: 0.01
  355. }
  356. }
  357. }
  358. layer {
  359. name: "rect_fire5_squeeze"
  360. type: "ReLU"
  361. relu_param {
  362. negative_slope: 0.01
  363. }
  364. bottom: "fire5_squeeze"
  365. top: "fire5_squeeze"
  366. }
  367. layer {
  368. name: "fire5_expand_1x1"
  369. type: "Convolution"
  370. bottom: "fire5_squeeze"
  371. top: "fire5_expand_1x1"
  372. convolution_param {
  373. num_output: 128
  374. kernel_size: 1
  375. stride: 1
  376. pad: 0
  377. weight_filler {
  378. type: "xavier"
  379. }
  380. bias_filler {
  381. type: "constant"
  382. value: 0.01
  383. }
  384. }
  385. }
  386. layer {
  387. name: "rect_fire5_expand_1x1"
  388. type: "ReLU"
  389. relu_param {
  390. negative_slope: 0.01
  391. }
  392. bottom: "fire5_expand_1x1"
  393. top: "fire5_expand_1x1"
  394. }
  395. layer {
  396. name: "fire5_expand_3x3"
  397. type: "Convolution"
  398. bottom: "fire5_squeeze"
  399. top: "fire5_expand_3x3"
  400. convolution_param {
  401. num_output: 128
  402. kernel_size: 3
  403. stride: 1
  404. pad: 1
  405. weight_filler {
  406. type: "xavier"
  407. }
  408. bias_filler {
  409. type: "constant"
  410. value: 0.01
  411. }
  412. }
  413. }
  414. layer {
  415. name: "rect_fire5_expand_3x3"
  416. type: "ReLU"
  417. relu_param {
  418. negative_slope: 0.01
  419. }
  420. bottom: "fire5_expand_3x3"
  421. top: "fire5_expand_3x3"
  422. }
  423. layer {
  424. name: "fire5"
  425. type: "Concat"
  426. bottom: "fire5_expand_1x1"
  427. bottom: "fire5_expand_3x3"
  428. top: "fire5"
  429. concat_param {
  430. axis: 1
  431. }
  432. }
  433. layer {
  434. name: "fire6_squeeze"
  435. type: "Convolution"
  436. bottom: "fire5"
  437. top: "fire6_squeeze"
  438. convolution_param {
  439. num_output: 48
  440. kernel_size: 1
  441. stride: 1
  442. pad: 0
  443. weight_filler {
  444. type: "xavier"
  445. }
  446. bias_filler {
  447. type: "constant"
  448. value: 0.01
  449. }
  450. }
  451. }
  452. layer {
  453. name: "rect_fire6_squeeze"
  454. type: "ReLU"
  455. relu_param {
  456. negative_slope: 0.01
  457. }
  458. bottom: "fire6_squeeze"
  459. top: "fire6_squeeze"
  460. }
  461. layer {
  462. name: "fire6_expand_1x1"
  463. type: "Convolution"
  464. bottom: "fire6_squeeze"
  465. top: "fire6_expand_1x1"
  466. convolution_param {
  467. num_output: 192
  468. kernel_size: 1
  469. stride: 1
  470. pad: 0
  471. weight_filler {
  472. type: "xavier"
  473. }
  474. bias_filler {
  475. type: "constant"
  476. value: 0.01
  477. }
  478. }
  479. }
  480. layer {
  481. name: "rect_fire6_expand_1x1"
  482. type: "ReLU"
  483. relu_param {
  484. negative_slope: 0.01
  485. }
  486. bottom: "fire6_expand_1x1"
  487. top: "fire6_expand_1x1"
  488. }
  489. layer {
  490. name: "fire6_expand_3x3"
  491. type: "Convolution"
  492. bottom: "fire6_squeeze"
  493. top: "fire6_expand_3x3"
  494. convolution_param {
  495. num_output: 192
  496. kernel_size: 3
  497. stride: 1
  498. pad: 1
  499. weight_filler {
  500. type: "xavier"
  501. }
  502. bias_filler {
  503. type: "constant"
  504. value: 0.01
  505. }
  506. }
  507. }
  508. layer {
  509. name: "rect_fire6_expand_3x3"
  510. type: "ReLU"
  511. relu_param {
  512. negative_slope: 0.01
  513. }
  514. bottom: "fire6_expand_3x3"
  515. top: "fire6_expand_3x3"
  516. }
  517. layer {
  518. name: "fire6"
  519. type: "Concat"
  520. bottom: "fire6_expand_1x1"
  521. bottom: "fire6_expand_3x3"
  522. top: "fire6"
  523. concat_param {
  524. axis: 1
  525. }
  526. }
  527. layer {
  528. name: "fire7_squeeze"
  529. type: "Convolution"
  530. bottom: "fire6"
  531. top: "fire7_squeeze"
  532. convolution_param {
  533. num_output: 48
  534. kernel_size: 1
  535. stride: 1
  536. pad: 0
  537. weight_filler {
  538. type: "xavier"
  539. }
  540. bias_filler {
  541. type: "constant"
  542. value: 0.01
  543. }
  544. }
  545. }
  546. layer {
  547. name: "rect_fire7_squeeze"
  548. type: "ReLU"
  549. relu_param {
  550. negative_slope: 0.01
  551. }
  552. bottom: "fire7_squeeze"
  553. top: "fire7_squeeze"
  554. }
  555. layer {
  556. name: "fire7_expand_1x1"
  557. type: "Convolution"
  558. bottom: "fire7_squeeze"
  559. top: "fire7_expand_1x1"
  560. convolution_param {
  561. num_output: 192
  562. kernel_size: 1
  563. stride: 1
  564. pad: 0
  565. weight_filler {
  566. type: "xavier"
  567. }
  568. bias_filler {
  569. type: "constant"
  570. value: 0.01
  571. }
  572. }
  573. }
  574. layer {
  575. name: "rect_fire7_expand_1x1"
  576. type: "ReLU"
  577. relu_param {
  578. negative_slope: 0.01
  579. }
  580. bottom: "fire7_expand_1x1"
  581. top: "fire7_expand_1x1"
  582. }
  583. layer {
  584. name: "fire7_expand_3x3"
  585. type: "Convolution"
  586. bottom: "fire7_squeeze"
  587. top: "fire7_expand_3x3"
  588. convolution_param {
  589. num_output: 192
  590. kernel_size: 3
  591. stride: 1
  592. pad: 1
  593. weight_filler {
  594. type: "xavier"
  595. }
  596. bias_filler {
  597. type: "constant"
  598. value: 0.01
  599. }
  600. }
  601. }
  602. layer {
  603. name: "rect_fire7_expand_3x3"
  604. type: "ReLU"
  605. relu_param {
  606. negative_slope: 0.01
  607. }
  608. bottom: "fire7_expand_3x3"
  609. top: "fire7_expand_3x3"
  610. }
  611. layer {
  612. name: "fire7"
  613. type: "Concat"
  614. bottom: "fire7_expand_1x1"
  615. bottom: "fire7_expand_3x3"
  616. top: "fire7"
  617. concat_param {
  618. axis: 1
  619. }
  620. }
  621. layer {
  622. name: "fire8_squeeze"
  623. type: "Convolution"
  624. bottom: "fire7"
  625. top: "fire8_squeeze"
  626. convolution_param {
  627. num_output: 64
  628. kernel_size: 1
  629. stride: 1
  630. pad: 0
  631. weight_filler {
  632. type: "xavier"
  633. }
  634. bias_filler {
  635. type: "constant"
  636. value: 0.01
  637. }
  638. }
  639. }
  640. layer {
  641. name: "rect_fire8_squeeze"
  642. type: "ReLU"
  643. relu_param {
  644. negative_slope: 0.01
  645. }
  646. bottom: "fire8_squeeze"
  647. top: "fire8_squeeze"
  648. }
  649. layer {
  650. name: "fire8_expand_1x1"
  651. type: "Convolution"
  652. bottom: "fire8_squeeze"
  653. top: "fire8_expand_1x1"
  654. convolution_param {
  655. num_output: 256
  656. kernel_size: 1
  657. stride: 1
  658. pad: 0
  659. weight_filler {
  660. type: "xavier"
  661. }
  662. bias_filler {
  663. type: "constant"
  664. value: 0.01
  665. }
  666. }
  667. }
  668. layer {
  669. name: "rect_fire8_expand_1x1"
  670. type: "ReLU"
  671. relu_param {
  672. negative_slope: 0.01
  673. }
  674. bottom: "fire8_expand_1x1"
  675. top: "fire8_expand_1x1"
  676. }
  677. layer {
  678. name: "fire8_expand_3x3"
  679. type: "Convolution"
  680. bottom: "fire8_squeeze"
  681. top: "fire8_expand_3x3"
  682. convolution_param {
  683. num_output: 256
  684. kernel_size: 3
  685. stride: 1
  686. pad: 1
  687. weight_filler {
  688. type: "xavier"
  689. }
  690. bias_filler {
  691. type: "constant"
  692. value: 0.01
  693. }
  694. }
  695. }
  696. layer {
  697. name: "rect_fire8_expand_3x3"
  698. type: "ReLU"
  699. relu_param {
  700. negative_slope: 0.01
  701. }
  702. bottom: "fire8_expand_3x3"
  703. top: "fire8_expand_3x3"
  704. }
  705. layer {
  706. name: "fire8"
  707. type: "Concat"
  708. bottom: "fire8_expand_1x1"
  709. bottom: "fire8_expand_3x3"
  710. top: "fire8"
  711. concat_param {
  712. axis: 1
  713. }
  714. }
  715. layer {
  716. name: "pool8"
  717. type: "Pooling"
  718. bottom: "fire8"
  719. top: "pool8"
  720. pooling_param {
  721. pool: MAX
  722. kernel_size: 3
  723. stride: 2
  724. }
  725. }
  726. layer {
  727. name: "fire9_squeeze"
  728. type: "Convolution"
  729. bottom: "pool8"
  730. top: "fire9_squeeze"
  731. convolution_param {
  732. num_output: 64
  733. kernel_size: 1
  734. stride: 1
  735. pad: 0
  736. weight_filler {
  737. type: "xavier"
  738. }
  739. bias_filler {
  740. type: "constant"
  741. value: 0.01
  742. }
  743. }
  744. }
  745. layer {
  746. name: "rect_fire9_squeeze"
  747. type: "ReLU"
  748. relu_param {
  749. negative_slope: 0.01
  750. }
  751. bottom: "fire9_squeeze"
  752. top: "fire9_squeeze"
  753. }
  754. layer {
  755. name: "fire9_expand_1x1"
  756. type: "Convolution"
  757. bottom: "fire9_squeeze"
  758. top: "fire9_expand_1x1"
  759. convolution_param {
  760. num_output: 256
  761. kernel_size: 1
  762. stride: 1
  763. pad: 0
  764. weight_filler {
  765. type: "xavier"
  766. }
  767. bias_filler {
  768. type: "constant"
  769. value: 0.01
  770. }
  771. }
  772. }
  773. layer {
  774. name: "rect_fire9_expand_1x1"
  775. type: "ReLU"
  776. relu_param {
  777. negative_slope: 0.01
  778. }
  779. bottom: "fire9_expand_1x1"
  780. top: "fire9_expand_1x1"
  781. }
  782. layer {
  783. name: "fire9_expand_3x3"
  784. type: "Convolution"
  785. bottom: "fire9_squeeze"
  786. top: "fire9_expand_3x3"
  787. convolution_param {
  788. num_output: 256
  789. kernel_size: 3
  790. stride: 1
  791. pad: 1
  792. weight_filler {
  793. type: "xavier"
  794. }
  795. bias_filler {
  796. type: "constant"
  797. value: 0.01
  798. }
  799. }
  800. }
  801. layer {
  802. name: "rect_fire9_expand_3x3"
  803. type: "ReLU"
  804. relu_param {
  805. negative_slope: 0.01
  806. }
  807. bottom: "fire9_expand_3x3"
  808. top: "fire9_expand_3x3"
  809. }
  810. layer {
  811. name: "fire9"
  812. type: "Concat"
  813. bottom: "fire9_expand_1x1"
  814. bottom: "fire9_expand_3x3"
  815. top: "fire9"
  816. concat_param {
  817. axis: 1
  818. }
  819. }
  820. layer {
  821. name: "conv10"
  822. type: "Convolution"
  823. bottom: "fire9"
  824. top: "conv10"
  825. convolution_param {
  826. num_output: 1000
  827. kernel_size: 1
  828. stride: 1
  829. weight_filler {
  830. type: "gaussian"
  831. mean: 0.0
  832. std: 0.01
  833. }
  834. bias_filler {
  835. type: "constant"
  836. value: 0.01
  837. }
  838. }
  839. }
  840. layer {
  841. name: "rect_conv10"
  842. type: "ReLU"
  843. relu_param {
  844. negative_slope: 0.01
  845. }
  846. bottom: "conv10"
  847. top: "conv10"
  848. }
  849. layer {
  850. name: "pool10"
  851. type: "Pooling"
  852. bottom: "conv10"
  853. top: "pool10"
  854. pooling_param {
  855. pool: AVE
  856. global_pooling: true
  857. }
  858. }
  859. layer {
  860. name: "predictions"
  861. type: "Softmax"
  862. bottom: "pool10"
  863. top: "predictions"
  864. softmax_param {
  865. axis: 1
  866. }
  867. }