Part-Aware Data Augmentation for 3D Object Detection in Point Cloud

This repository contains a reference implementation of our Part-Aware Data Augmentation for 3D Object Detection in Point Cloud.

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

@article{choi2020part,
  title={Part-Aware Data Augmentation for 3D Object Detection in Point Cloud},
  author={Choi, Jaeseok and Song, Yeji and Kwak, Nojun},
  journal={arXiv preprint arXiv:2007.13373},
  year={2020}
}

Prerequisites

Our code was tested on second.pytorch and OpenPCDet.
This repository contains only part-aware data augmentation code.
Refer to the link above for code such as data loader or detector.

Usage

Args:
    ** only supports KITTI format **
    points: lidar points (N, 4), 
    gt_boxes: ground truth boxes (B, 7),
    gt_names: ground truth classes (B, 1), 
    class_names: list of classes to augment (3),
    pa_aug_param: parameters for PA_AUG (string).

Returns:
    points: augmented lidar points (N', 4),
    gt_boxes_mask: mask for gt_boxes (B)


class_names = ['Car', 'Pedestrian', 'Cyclist']
pa_aug_param = "dropout_p02_swap_p02_mix_p02_sparse40_p01_noise10_p01"

pa_aug = PartAwareAugmentation(points, gt_boxes, gt_names, class_names=class_names)
points, gt_boxes_mask = pa_aug.augment(pa_aug_param=pa_aug_param)
gt_boxes = gt_boxes[gt_boxes_mask]

Part-Aware Data Augmentation for 3D Object Detection in Point Cloud

Related tags

Overview

Part-Aware Data Augmentation for 3D Object Detection in Point Cloud

Prerequisites

Usage

Owner

Jaeseok Choi

Implementation of 'X-Linear Attention Networks for Image Captioning' [CVPR 2020]

Facial Expression Detection In The Realtime

A Simple LSTM-Based Solution for "Heartbeat Signal Classification and Prediction" in Tianchi

This is the code related to "Sparse-to-dense Feature Matching: Intra and Inter domain Cross-modal Learning in Domain Adaptation for 3D Semantic Segmentation" (ICCV 2021).

Citation Intent Classification in scientific papers using the Scicite dataset an Pytorch

covid question answering datasets and fine tuned models

Supplementary code for the paper "Meta-Solver for Neural Ordinary Differential Equations" https://arxiv.org/abs/2103.08561

An algorithm that handles large-scale aerial photo co-registration, based on SURF, RANSAC and PyTorch autograd.

In this project, we'll be making our own screen recorder in Python using some libraries.

NeRF visualization library under construction

This repository contains the code for our paper VDA (public in EMNLP2021 main conference)

Explainability for Vision Transformers (in PyTorch)

Multispectral Object Detection with Yolov5

Python-based Informatics Kit for Analysing Chemical Units

Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition

Learning 3D Part Assembly from a Single Image

Python/Rust implementations and notes from Proofs Arguments and Zero Knowledge

Deep and online learning with spiking neural networks in Python

Aircraft design optimization made fast through modern automatic differentiation

TorchPQ is a python library for Approximate Nearest Neighbor Search (ANNS) and Maximum Inner Product Search (MIPS) on GPU using Product Quantization (PQ) algorithm.