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

Official implementation of "StyleCariGAN: Caricature Generation via StyleGAN Feature Map Modulation" (SIGGRAPH 2021)

Python code to generate art with Generative Adversarial Network

PyTorch Implementation of Vector Quantized Variational AutoEncoders.

Python script that analyses the given datasets and comes up with the best polynomial regression representation with the smallest polynomial degree possible

Implicit Graph Neural Networks

rliable is an open-source Python library for reliable evaluation, even with a handful of runs, on reinforcement learning and machine learnings benchmarks.

HHP-Net: A light Heteroscedastic neural network for Head Pose estimation with uncertainty

Unoffical reMarkable AddOn for Firefox.

Global Pooling, More than Meets the Eye: Position Information is Encoded Channel-Wise in CNNs, ICCV 2021

UNet model with VGG11 encoder pre-trained on Kaggle Carvana dataset

Code for models used in Bashiri et al., "A Flow-based latent state generative model of neural population responses to natural images".

More Photos are All You Need: Semi-Supervised Learning for Fine-Grained Sketch Based Image Retrieval

A commany has recently introduced a new type of bidding, the average bidding, as an alternative to the bid given to the current maximum bidding

Official code for ICCV2021 paper "M3D-VTON: A Monocular-to-3D Virtual Try-on Network"

Tool for live presentations using manim

A resource for learning about ML, DL, PyTorch and TensorFlow. Feedback always appreciated :)

Implementation of: "Exploring Randomly Wired Neural Networks for Image Recognition"

This is a collection of our NAS and Vision Transformer work.

An end-to-end regression problem of predicting the price of properties in Bangalore.

Code for the paper: Fighting Fake News: Image Splice Detection via Learned Self-Consistency