Skip to content

Pytorch implementation for the paper: "FMFNet: Improve the 3D Object Detection and Tracking via Feature Map Flow" [IJCNN-2022]

License

Notifications You must be signed in to change notification settings

YoushaaMurhij/FMFNet

Repository files navigation

Official Pytorch implementation for "FMFNet: Improve the 3D Object Detection and Tracking via Feature Map Flow" Paper [IJCNN-2022]

workflow License: MIT Visitor count

FMF main blocks:

drawing

FMF Demo:

drawing

News

[03-10-2022] FMFNet Paper is published on IEEE Xplore 🔥
[22-07-2022] FMFNet Poster is represented in the International Joint Conference on Neural Networks (IJCNN 2022).
[26-04-2022] FMFNet is accepted at IEEE WCCI (IJCNN) 2022 🔥
[20-05-2021] FMFNet team got the 5th rank in Real-time 3D detection challenge hosted by CVPR 2021 Workshop on Autonomous Driving (WAD) and a brief technical report was published in the proceddings of the workshop.

Abstract

Three-dimensional object detection and tracking from point clouds are important computer vision tasks for robots and vehicles where objects can be represented as 3D boxes. Improving the accuracy of understanding the environment is critical for successful autonomous driving. This paper presents a simple yet efficient method called “Feature Map Flow, FMF” for 3D object detection and tracking, considering time-spatial feature map aggregation from different timesteps of deep neural model inference. Several versions of the FMF are proposed: from common concatenation to context-based feature map fusion and odometry usage for previous feature map affine transform. The proposed approach significantly improves the quality of 3D detection and tracking baseline on the nuScenes and Waymo benchmarks. The software implementation of the proposed method has been carried out for the server platform and the NVidia Jetson AGX Xavier single-board computer. Its results have demonstrated high application prospects both for automated 3D point cloud labeling and for embedded on-board vehicle systems.

Main Results

3D detection on nuScenes test set

Model MAP ↑ NDS ↑ FPS ↑
VoxelNet 58.0 65.9 17
PointPillars 53.8 62.7 29

3D Tracking on nuScenes test set

Model AMOTA ↑ IDS ↓
VoxelNet 61.2 870
PointPillars 58.1 736

3D detection on Waymo test set

Model Veh ↑ Ped ↑ Cyc ↑ mAPH ↑ Latency,ms
VoxelNet 70.74 65.46 67.63 67.95 86.11
PointPillars (FP16) 69.65 54.61 62.28 62.18 62.30

3D detection on Waymo Val set

Model Veh ↑ Ped ↑ Cyc ↑ mAPH ↑ Latency,ms
VoxelNet 66.68 63.62 68.64 65.98 86.11
VoxelNet (FP16) 66.68 63.06 67.24 65.85 77.53
PointPillars 62.35 59.67 66.71 62.43 82.06
PointPillars (FP16) 61.75 58.12 65.26 62.19 62.30

All results are tested on a RTX 3060 ti GPU with batch size 1.

Use FMFNet

Follow the provided steps to reproduce our results on nuScenes validation and test sets and get pretrained models.

Please refer to INSTALL to run the docker container for FMFNet. For training and testing on nuScenes, please follow the instructions in START. For WAYMO dataset, you can check START_WAYMO

Lisence

FMFNet is released under MIT license (see LICENSE). It is developed based on a forked version of CenterPoint. We also used code from det3d, CenterNet and CenterTrack.

Contact

Questions and suggestions are welcome!

Youshaa Murhij yosha[dot]morheg[at]phystech[dot]edu
Dmitry Yudin yudin[dot]da[at]mipt[dot]ru

Citation

If you find this project useful in your research, please consider citing our work:

@inproceedings{murhij2022fmfnet,
  title={FMFNet: Improve the 3D Object Detection and Tracking via Feature Map Flow},
  author={Murhij, Youshaa and Yudin, Dmitry},
  booktitle={2022 International Joint Conference on Neural Networks (IJCNN)},
  pages={1--8},
  year={2022},
  organization={IEEE}
}

About

Pytorch implementation for the paper: "FMFNet: Improve the 3D Object Detection and Tracking via Feature Map Flow" [IJCNN-2022]

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published