Paper ID | IMT-1.6 | ||
Paper Title | ReinforceDet: Object Detection by Integrating Reinforcement Learning with Decoupled Pipeline | ||
Authors | Man Zhou, University of Science and Technology of China, China; Liu Liu, Shanghai Jiao Tong University, China; Rujing Wang, Chinese Academy of Sciences, China | ||
Session | IMT-1: Computational Imaging Learning-based Models | ||
Location | Area J | ||
Session Time: | Tuesday, 21 September, 08:00 - 09:30 | ||
Presentation Time: | Tuesday, 21 September, 08:00 - 09:30 | ||
Presentation | Poster | ||
Topic | Computational Imaging Methods and Models: Learning-Based Models | ||
IEEE Xplore Open Preview | Click here to view in IEEE Xplore | ||
Abstract | Recent object detection methods largely rely on numerous pre-defined anchors that suffer from huge computational cost and resource consumption. To solve this issue, we propose a low-memory deep reinforcement learning based anchor-free object detection approach, namely ReinforceDet, which computes few but accurate region proposals for detection. Specifically, the extracted feature maps are fed into a reinforcement learning network to localize objects as initial region proposals with our re-designed reward function and then adopt another neural network to refine them. To speed up this process in test phase, we decouple the two-branch CNN networks as light-head cascaded subnetworks, named IoU-net and bounding box net. Experimental results show that ReinforceDet could obtain the state-of-the-art performance with much lower compitational and memory cost. |