API tools faq
paste
Guest User

Untitled

a guest
Dec 10th, 2018
75
0
Never
Add comment
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 15.20 KB | None | 0 0
raw download clone embed print report
  1. name: "cnn_fd_004_sq_light_ssd"
  2. layer {
  3. name: "data"
  4. type: "Input"
  5. top: "data"
  6. input_param {
  7. shape {
  8. dim: 1
  9. dim: 3
  10. dim: 300
  11. dim: 300
  12. }
  13. }
  14. }
  15. layer {
  16. name: "Scale1"
  17. type: "Scale"
  18. bottom: "data"
  19. top: "data"
  20. param {
  21. lr_mult: 1.0
  22. decay_mult: 1.0
  23. }
  24. param {
  25. lr_mult: 2.0
  26. decay_mult: 1.0
  27. }
  28. scale_param {
  29. bias_term: true
  30. }
  31. }
  32. layer {
  33. name: "Convolution1"
  34. type: "Convolution"
  35. bottom: "data"
  36. top: "Convolution1"
  37. convolution_param {
  38. num_output: 32
  39. pad: 3
  40. kernel_size: 7
  41. stride: 2
  42. weight_filler {
  43. type: "xavier"
  44. }
  45. }
  46. }
  47. layer {
  48. name: "ReLU1"
  49. type: "ReLU"
  50. bottom: "Convolution1"
  51. top: "Convolution1"
  52. }
  53. layer {
  54. name: "Pooling1"
  55. type: "Pooling"
  56. bottom: "Convolution1"
  57. top: "Pooling1"
  58. pooling_param {
  59. pool: MAX
  60. kernel_size: 3
  61. stride: 2
  62. }
  63. }
  64. layer {
  65. name: "fire2/squeeze1x1"
  66. type: "Convolution"
  67. bottom: "Pooling1"
  68. top: "fire2/squeeze1x1"
  69. convolution_param {
  70. num_output: 8
  71. kernel_size: 1
  72. stride: 1
  73. weight_filler {
  74. type: "xavier"
  75. }
  76. }
  77. }
  78. layer {
  79. name: "fire2/relu_squeeze1x1"
  80. type: "ReLU"
  81. bottom: "fire2/squeeze1x1"
  82. top: "fire2/squeeze1x1"
  83. }
  84. layer {
  85. name: "fire2/expand1x1"
  86. type: "Convolution"
  87. bottom: "fire2/squeeze1x1"
  88. top: "fire2/expand1x1"
  89. convolution_param {
  90. num_output: 32
  91. kernel_size: 1
  92. weight_filler {
  93. type: "xavier"
  94. }
  95. }
  96. }
  97. layer {
  98. name: "fire2/relu_expand1x1"
  99. type: "ReLU"
  100. bottom: "fire2/expand1x1"
  101. top: "fire2/expand1x1"
  102. }
  103. layer {
  104. name: "fire2/expand3x3"
  105. type: "Convolution"
  106. bottom: "fire2/squeeze1x1"
  107. top: "fire2/expand3x3"
  108. convolution_param {
  109. num_output: 32
  110. pad: 1
  111. kernel_size: 3
  112. weight_filler {
  113. type: "xavier"
  114. }
  115. }
  116. }
  117. layer {
  118. name: "fire2/relu_expand3x3"
  119. type: "ReLU"
  120. bottom: "fire2/expand3x3"
  121. top: "fire2/expand3x3"
  122. }
  123. layer {
  124. name: "Concat1"
  125. type: "Concat"
  126. bottom: "fire2/expand3x3"
  127. bottom: "fire2/expand1x1"
  128. top: "Concat1"
  129. concat_param {
  130. axis: 1
  131. }
  132. }
  133. layer {
  134. name: "fire3/squeeze1x1"
  135. type: "Convolution"
  136. bottom: "Concat1"
  137. top: "fire3/squeeze1x1"
  138. convolution_param {
  139. num_output: 8
  140. kernel_size: 1
  141. stride: 1
  142. weight_filler {
  143. type: "xavier"
  144. }
  145. }
  146. }
  147. layer {
  148. name: "fire3/relu_squeeze1x1"
  149. type: "ReLU"
  150. bottom: "fire3/squeeze1x1"
  151. top: "fire3/squeeze1x1"
  152. }
  153. layer {
  154. name: "fire3/expand1x1"
  155. type: "Convolution"
  156. bottom: "fire3/squeeze1x1"
  157. top: "fire3/expand1x1"
  158. convolution_param {
  159. num_output: 32
  160. kernel_size: 1
  161. weight_filler {
  162. type: "xavier"
  163. }
  164. }
  165. }
  166. layer {
  167. name: "fire3/relu_expand1x1"
  168. type: "ReLU"
  169. bottom: "fire3/expand1x1"
  170. top: "fire3/expand1x1"
  171. }
  172. layer {
  173. name: "fire3/expand3x3"
  174. type: "Convolution"
  175. bottom: "fire3/squeeze1x1"
  176. top: "fire3/expand3x3"
  177. convolution_param {
  178. num_output: 32
  179. pad: 1
  180. kernel_size: 3
  181. weight_filler {
  182. type: "xavier"
  183. }
  184. }
  185. }
  186. layer {
  187. name: "fire3/relu_expand3x3"
  188. type: "ReLU"
  189. bottom: "fire3/expand3x3"
  190. top: "fire3/expand3x3"
  191. }
  192. layer {
  193. name: "Concat2"
  194. type: "Concat"
  195. bottom: "fire3/expand3x3"
  196. bottom: "fire3/expand1x1"
  197. top: "Concat2"
  198. concat_param {
  199. axis: 1
  200. }
  201. }
  202. layer {
  203. name: "fire4/squeeze1x1"
  204. type: "Convolution"
  205. bottom: "Concat2"
  206. top: "fire4/squeeze1x1"
  207. convolution_param {
  208. num_output: 16
  209. kernel_size: 1
  210. stride: 1
  211. weight_filler {
  212. type: "xavier"
  213. }
  214. }
  215. }
  216. layer {
  217. name: "fire4/relu_squeeze1x1"
  218. type: "ReLU"
  219. bottom: "fire4/squeeze1x1"
  220. top: "fire4/squeeze1x1"
  221. }
  222. layer {
  223. name: "fire4/expand1x1"
  224. type: "Convolution"
  225. bottom: "fire4/squeeze1x1"
  226. top: "fire4/expand1x1"
  227. convolution_param {
  228. num_output: 64
  229. kernel_size: 1
  230. weight_filler {
  231. type: "xavier"
  232. }
  233. }
  234. }
  235. layer {
  236. name: "fire4/relu_expand1x1"
  237. type: "ReLU"
  238. bottom: "fire4/expand1x1"
  239. top: "fire4/expand1x1"
  240. }
  241. layer {
  242. name: "fire4/expand3x3"
  243. type: "Convolution"
  244. bottom: "fire4/squeeze1x1"
  245. top: "fire4/expand3x3"
  246. convolution_param {
  247. num_output: 64
  248. pad: 1
  249. kernel_size: 3
  250. weight_filler {
  251. type: "xavier"
  252. }
  253. }
  254. }
  255. layer {
  256. name: "fire4/relu_expand3x3"
  257. type: "ReLU"
  258. bottom: "fire4/expand3x3"
  259. top: "fire4/expand3x3"
  260. }
  261. layer {
  262. name: "Concat3"
  263. type: "Concat"
  264. bottom: "fire4/expand3x3"
  265. bottom: "fire4/expand1x1"
  266. top: "Concat3"
  267. concat_param {
  268. axis: 1
  269. }
  270. }
  271. layer {
  272. name: "fire5/squeeze1x1"
  273. type: "Convolution"
  274. bottom: "Concat3"
  275. top: "fire5/squeeze1x1"
  276. convolution_param {
  277. num_output: 16
  278. kernel_size: 1
  279. stride: 2
  280. weight_filler {
  281. type: "xavier"
  282. }
  283. }
  284. }
  285. layer {
  286. name: "fire5/relu_squeeze1x1"
  287. type: "ReLU"
  288. bottom: "fire5/squeeze1x1"
  289. top: "fire5/squeeze1x1"
  290. }
  291. layer {
  292. name: "fire5/expand1x1"
  293. type: "Convolution"
  294. bottom: "fire5/squeeze1x1"
  295. top: "fire5/expand1x1"
  296. convolution_param {
  297. num_output: 64
  298. kernel_size: 1
  299. weight_filler {
  300. type: "xavier"
  301. }
  302. }
  303. }
  304. layer {
  305. name: "fire5/relu_expand1x1"
  306. type: "ReLU"
  307. bottom: "fire5/expand1x1"
  308. top: "fire5/expand1x1"
  309. }
  310. layer {
  311. name: "fire5/expand3x3"
  312. type: "Convolution"
  313. bottom: "fire5/squeeze1x1"
  314. top: "fire5/expand3x3"
  315. convolution_param {
  316. num_output: 64
  317. pad: 1
  318. kernel_size: 3
  319. weight_filler {
  320. type: "xavier"
  321. }
  322. }
  323. }
  324. layer {
  325. name: "fire5/relu_expand3x3"
  326. type: "ReLU"
  327. bottom: "fire5/expand3x3"
  328. top: "fire5/expand3x3"
  329. }
  330. layer {
  331. name: "Concat4"
  332. type: "Concat"
  333. bottom: "fire5/expand3x3"
  334. bottom: "fire5/expand1x1"
  335. top: "Concat4"
  336. concat_param {
  337. axis: 1
  338. }
  339. }
  340. layer {
  341. name: "fire6/squeeze1x1"
  342. type: "Convolution"
  343. bottom: "Concat4"
  344. top: "fire6/squeeze1x1"
  345. convolution_param {
  346. num_output: 24
  347. kernel_size: 1
  348. stride: 1
  349. weight_filler {
  350. type: "xavier"
  351. }
  352. }
  353. }
  354. layer {
  355. name: "fire6/relu_squeeze1x1"
  356. type: "ReLU"
  357. bottom: "fire6/squeeze1x1"
  358. top: "fire6/squeeze1x1"
  359. }
  360. layer {
  361. name: "fire6/expand1x1"
  362. type: "Convolution"
  363. bottom: "fire6/squeeze1x1"
  364. top: "fire6/expand1x1"
  365. convolution_param {
  366. num_output: 96
  367. kernel_size: 1
  368. weight_filler {
  369. type: "xavier"
  370. }
  371. }
  372. }
  373. layer {
  374. name: "fire6/relu_expand1x1"
  375. type: "ReLU"
  376. bottom: "fire6/expand1x1"
  377. top: "fire6/expand1x1"
  378. }
  379. layer {
  380. name: "fire6/expand3x3"
  381. type: "Convolution"
  382. bottom: "fire6/squeeze1x1"
  383. top: "fire6/expand3x3"
  384. convolution_param {
  385. num_output: 96
  386. pad: 1
  387. kernel_size: 3
  388. weight_filler {
  389. type: "xavier"
  390. }
  391. }
  392. }
  393. layer {
  394. name: "fire6/relu_expand3x3"
  395. type: "ReLU"
  396. bottom: "fire6/expand3x3"
  397. top: "fire6/expand3x3"
  398. }
  399. layer {
  400. name: "Concat5"
  401. type: "Concat"
  402. bottom: "fire6/expand3x3"
  403. bottom: "fire6/expand1x1"
  404. top: "Concat5"
  405. concat_param {
  406. axis: 1
  407. }
  408. }
  409. layer {
  410. name: "fire7/squeeze1x1"
  411. type: "Convolution"
  412. bottom: "Concat5"
  413. top: "fire7/squeeze1x1"
  414. convolution_param {
  415. num_output: 24
  416. kernel_size: 1
  417. stride: 1
  418. weight_filler {
  419. type: "xavier"
  420. }
  421. }
  422. }
  423. layer {
  424. name: "fire7/relu_squeeze1x1"
  425. type: "ReLU"
  426. bottom: "fire7/squeeze1x1"
  427. top: "fire7/squeeze1x1"
  428. }
  429. layer {
  430. name: "fire7/expand1x1"
  431. type: "Convolution"
  432. bottom: "fire7/squeeze1x1"
  433. top: "fire7/expand1x1"
  434. convolution_param {
  435. num_output: 96
  436. kernel_size: 1
  437. weight_filler {
  438. type: "xavier"
  439. }
  440. }
  441. }
  442. layer {
  443. name: "fire7/relu_expand1x1"
  444. type: "ReLU"
  445. bottom: "fire7/expand1x1"
  446. top: "fire7/expand1x1"
  447. }
  448. layer {
  449. name: "fire7/expand3x3"
  450. type: "Convolution"
  451. bottom: "fire7/squeeze1x1"
  452. top: "fire7/expand3x3"
  453. convolution_param {
  454. num_output: 96
  455. pad: 1
  456. kernel_size: 3
  457. weight_filler {
  458. type: "xavier"
  459. }
  460. }
  461. }
  462. layer {
  463. name: "fire7/relu_expand3x3"
  464. type: "ReLU"
  465. bottom: "fire7/expand3x3"
  466. top: "fire7/expand3x3"
  467. }
  468. layer {
  469. name: "Concat6"
  470. type: "Concat"
  471. bottom: "fire7/expand3x3"
  472. bottom: "fire7/expand1x1"
  473. top: "Concat6"
  474. concat_param {
  475. axis: 1
  476. }
  477. }
  478. layer {
  479. name: "fire8/squeeze1x1"
  480. type: "Convolution"
  481. bottom: "Concat6"
  482. top: "fire8/squeeze1x1"
  483. convolution_param {
  484. num_output: 32
  485. kernel_size: 1
  486. stride: 1
  487. weight_filler {
  488. type: "xavier"
  489. }
  490. }
  491. }
  492. layer {
  493. name: "fire8/relu_squeeze1x1"
  494. type: "ReLU"
  495. bottom: "fire8/squeeze1x1"
  496. top: "fire8/squeeze1x1"
  497. }
  498. layer {
  499. name: "fire8/expand1x1"
  500. type: "Convolution"
  501. bottom: "fire8/squeeze1x1"
  502. top: "fire8/expand1x1"
  503. convolution_param {
  504. num_output: 128
  505. kernel_size: 1
  506. weight_filler {
  507. type: "xavier"
  508. }
  509. }
  510. }
  511. layer {
  512. name: "fire8/relu_expand1x1"
  513. type: "ReLU"
  514. bottom: "fire8/expand1x1"
  515. top: "fire8/expand1x1"
  516. }
  517. layer {
  518. name: "fire8/expand3x3"
  519. type: "Convolution"
  520. bottom: "fire8/squeeze1x1"
  521. top: "fire8/expand3x3"
  522. convolution_param {
  523. num_output: 128
  524. pad: 1
  525. kernel_size: 3
  526. weight_filler {
  527. type: "xavier"
  528. }
  529. }
  530. }
  531. layer {
  532. name: "fire8/relu_expand3x3"
  533. type: "ReLU"
  534. bottom: "fire8/expand3x3"
  535. top: "fire8/expand3x3"
  536. }
  537. layer {
  538. name: "Concat7"
  539. type: "Concat"
  540. bottom: "fire8/expand3x3"
  541. bottom: "fire8/expand1x1"
  542. top: "Concat7"
  543. concat_param {
  544. axis: 1
  545. }
  546. }
  547. layer {
  548. name: "fire9/squeeze1x1"
  549. type: "Convolution"
  550. bottom: "Concat7"
  551. top: "fire9/squeeze1x1"
  552. convolution_param {
  553. num_output: 32
  554. kernel_size: 1
  555. stride: 2
  556. weight_filler {
  557. type: "xavier"
  558. }
  559. }
  560. }
  561. layer {
  562. name: "fire9/relu_squeeze1x1"
  563. type: "ReLU"
  564. bottom: "fire9/squeeze1x1"
  565. top: "fire9/squeeze1x1"
  566. }
  567. layer {
  568. name: "fire9/expand1x1"
  569. type: "Convolution"
  570. bottom: "fire9/squeeze1x1"
  571. top: "fire9/expand1x1"
  572. convolution_param {
  573. num_output: 128
  574. kernel_size: 1
  575. weight_filler {
  576. type: "xavier"
  577. }
  578. }
  579. }
  580. layer {
  581. name: "fire9/relu_expand1x1"
  582. type: "ReLU"
  583. bottom: "fire9/expand1x1"
  584. top: "fire9/expand1x1"
  585. }
  586. layer {
  587. name: "fire9/expand3x3"
  588. type: "Convolution"
  589. bottom: "fire9/squeeze1x1"
  590. top: "fire9/expand3x3"
  591. convolution_param {
  592. num_output: 128
  593. pad: 1
  594. kernel_size: 3
  595. weight_filler {
  596. type: "xavier"
  597. }
  598. }
  599. }
  600. layer {
  601. name: "fire9/relu_expand3x3"
  602. type: "ReLU"
  603. bottom: "fire9/expand3x3"
  604. top: "fire9/expand3x3"
  605. }
  606. layer {
  607. name: "Concat8"
  608. type: "Concat"
  609. bottom: "fire9/expand3x3"
  610. bottom: "fire9/expand1x1"
  611. top: "Concat8"
  612. concat_param {
  613. axis: 1
  614. }
  615. }
  616. layer {
  617. name: "fire10/squeeze1x1"
  618. type: "Convolution"
  619. bottom: "Concat8"
  620. top: "fire10/squeeze1x1"
  621. convolution_param {
  622. num_output: 48
  623. kernel_size: 1
  624. stride: 1
  625. weight_filler {
  626. type: "xavier"
  627. }
  628. }
  629. }
  630. layer {
  631. name: "fire10/relu_squeeze1x1"
  632. type: "ReLU"
  633. bottom: "fire10/squeeze1x1"
  634. top: "fire10/squeeze1x1"
  635. }
  636. layer {
  637. name: "fire10/expand1x1"
  638. type: "Convolution"
  639. bottom: "fire10/squeeze1x1"
  640. top: "fire10/expand1x1"
  641. convolution_param {
  642. num_output: 192
  643. kernel_size: 1
  644. weight_filler {
  645. type: "xavier"
  646. }
  647. }
  648. }
  649. layer {
  650. name: "fire10/relu_expand1x1"
  651. type: "ReLU"
  652. bottom: "fire10/expand1x1"
  653. top: "fire10/expand1x1"
  654. }
  655. layer {
  656. name: "fire10/expand3x3"
  657. type: "Convolution"
  658. bottom: "fire10/squeeze1x1"
  659. top: "fire10/expand3x3"
  660. convolution_param {
  661. num_output: 192
  662. pad: 1
  663. kernel_size: 3
  664. weight_filler {
  665. type: "xavier"
  666. }
  667. }
  668. }
  669. layer {
  670. name: "fire10/relu_expand3x3"
  671. type: "ReLU"
  672. bottom: "fire10/expand3x3"
  673. top: "fire10/expand3x3"
  674. }
  675. layer {
  676. name: "Concat9"
  677. type: "Concat"
  678. bottom: "fire10/expand3x3"
  679. bottom: "fire10/expand1x1"
  680. top: "Concat9"
  681. concat_param {
  682. axis: 1
  683. }
  684. }
  685. layer {
  686. name: "fire11/squeeze1x1"
  687. type: "Convolution"
  688. bottom: "Concat9"
  689. top: "fire11/squeeze1x1"
  690. convolution_param {
  691. num_output: 48
  692. kernel_size: 1
  693. stride: 1
  694. weight_filler {
  695. type: "xavier"
  696. }
  697. }
  698. }
  699. layer {
  700. name: "fire11/relu_squeeze1x1"
  701. type: "ReLU"
  702. bottom: "fire11/squeeze1x1"
  703. top: "fire11/squeeze1x1"
  704. }
  705. layer {
  706. name: "fire11/expand1x1"
  707. type: "Convolution"
  708. bottom: "fire11/squeeze1x1"
  709. top: "fire11/expand1x1"
  710. convolution_param {
  711. num_output: 192
  712. kernel_size: 1
  713. weight_filler {
  714. type: "xavier"
  715. }
  716. }
  717. }
  718. layer {
  719. name: "fire11/relu_expand1x1"
  720. type: "ReLU"
  721. bottom: "fire11/expand1x1"
  722. top: "fire11/expand1x1"
  723. }
  724. layer {
  725. name: "fire11/expand3x3"
  726. type: "Convolution"
  727. bottom: "fire11/squeeze1x1"
  728. top: "fire11/expand3x3"
  729. convolution_param {
  730. num_output: 192
  731. pad: 1
  732. kernel_size: 3
  733. weight_filler {
  734. type: "xavier"
  735. }
  736. }
  737. }
  738. layer {
  739. name: "fire11/relu_expand3x3"
  740. type: "ReLU"
  741. bottom: "fire11/expand3x3"
  742. top: "fire11/expand3x3"
  743. }
  744. layer {
  745. name: "Concat10"
  746. type: "Concat"
  747. bottom: "fire11/expand3x3"
  748. bottom: "fire11/expand1x1"
  749. top: "fc7"
  750. concat_param {
  751. axis: 1
  752. }
  753. }
  754. layer {
  755. name: "fc7_mbox_loc"
  756. type: "Convolution"
  757. bottom: "fc7"
  758. top: "fc7_mbox_loc"
  759. param {
  760. lr_mult: 1.0
  761. decay_mult: 1.0
  762. }
  763. param {
  764. lr_mult: 2.0
  765. decay_mult: 0.0
  766. }
  767. convolution_param {
  768. num_output: 36
  769. pad: 1
  770. kernel_size: 3
  771. stride: 1
  772. weight_filler {
  773. type: "xavier"
  774. }
  775. bias_filler {
  776. type: "constant"
  777. value: 0.0
  778. }
  779. }
  780. }
  781. layer {
  782. name: "fc7_mbox_loc_perm"
  783. type: "Permute"
  784. bottom: "fc7_mbox_loc"
  785. top: "fc7_mbox_loc_perm"
  786. permute_param {
  787. order: 0
  788. order: 2
  789. order: 3
  790. order: 1
  791. }
  792. }
  793. layer {
  794. name: "fc7_mbox_loc_flat"
  795. type: "Flatten"
  796. bottom: "fc7_mbox_loc_perm"
  797. top: "fc7_mbox_loc_flat"
  798. flatten_param {
  799. axis: 1
  800. }
  801. }
  802. layer {
  803. name: "fc7_mbox_conf"
  804. type: "Convolution"
  805. bottom: "fc7"
  806. top: "fc7_mbox_conf"
  807. param {
  808. lr_mult: 1.0
  809. decay_mult: 1.0
  810. }
  811. param {
  812. lr_mult: 2.0
  813. decay_mult: 0.0
  814. }
  815. convolution_param {
  816. num_output: 18
  817. pad: 1
  818. kernel_size: 3
  819. stride: 1
  820. weight_filler {
  821. type: "xavier"
  822. }
  823. bias_filler {
  824. type: "constant"
  825. value: 0.0
  826. }
  827. }
  828. }
  829. layer {
  830. name: "fc7_mbox_conf_perm"
  831. type: "Permute"
  832. bottom: "fc7_mbox_conf"
  833. top: "fc7_mbox_conf_perm"
  834. permute_param {
  835. order: 0
  836. order: 2
  837. order: 3
  838. order: 1
  839. }
  840. }
  841. layer {
  842. name: "fc7_mbox_conf_flat"
  843. type: "Flatten"
  844. bottom: "fc7_mbox_conf_perm"
  845. top: "fc7_mbox_conf_flat"
  846. flatten_param {
  847. axis: 1
  848. }
  849. }
  850. layer {
  851. name: "fc7_mbox_priorbox"
  852. type: "PriorBoxClustered"
  853. bottom: "fc7"
  854. bottom: "data"
  855. top: "fc7_mbox_priorbox"
  856. prior_box_param {
  857. clip: false
  858. variance: 0.1
  859. variance: 0.1
  860. variance: 0.2
  861. variance: 0.2
  862. step: 16.0
  863. offset: 0.5
  864. width: 9.4
  865. width: 25.1
  866. width: 14.7
  867. width: 34.7
  868. width: 143.0
  869. width: 77.4
  870. width: 128.8
  871. width: 51.1
  872. width: 75.6
  873. height: 15.0
  874. height: 39.6
  875. height: 25.5
  876. height: 63.2
  877. height: 227.5
  878. height: 162.9
  879. height: 124.5
  880. height: 105.1
  881. height: 72.6
  882. }
  883. }
  884. layer {
  885. name: "mbox_conf_reshape"
  886. type: "Reshape"
  887. bottom: "fc7_mbox_conf_flat"
  888. top: "mbox_conf_reshape"
  889. reshape_param {
  890. shape {
  891. dim: 0
  892. dim: -1
  893. dim: 2
  894. }
  895. }
  896. }
  897. layer {
  898. name: "mbox_conf_softmax"
  899. type: "Softmax"
  900. bottom: "mbox_conf_reshape"
  901. top: "mbox_conf_softmax"
  902. softmax_param {
  903. axis: 2
  904. }
  905. }
  906. layer {
  907. name: "mbox_conf_flatten"
  908. type: "Flatten"
  909. bottom: "mbox_conf_softmax"
  910. top: "mbox_conf_flatten"
  911. flatten_param {
  912. axis: 1
  913. }
  914. }
  915. layer {
  916. name: "detection_out"
  917. type: "DetectionOutput"
  918. bottom: "fc7_mbox_loc_flat"
  919. bottom: "mbox_conf_flatten"
  920. bottom: "fc7_mbox_priorbox"
  921. top: "detection_out"
  922. include {
  923. phase: TEST
  924. }
  925. detection_output_param {
  926. num_classes: 2
  927. share_location: true
  928. background_label_id: 0
  929. nms_param {
  930. nms_threshold: 0.45
  931. top_k: 400
  932. }
  933. code_type: CENTER_SIZE
  934. keep_top_k: 200
  935. confidence_threshold: 0.01
  936. }
  937. }
Add Comment
Please, Sign In to add comment
Public Pastes
We use cookies for various purposes including analytics. By continuing to use Pastebin, you agree to our use of cookies as described in the Cookies Policy.  OK, I Understand
Not a member of Pastebin yet?
Sign Up, it unlocks many cool features!