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

This is a Deep Leaning API for classifying emotions from human face and human audios.

Pytorch code for our paper Beyond ImageNet Attack: Towards Crafting Adversarial Examples for Black-box Domains)

DynaTune: Dynamic Tensor Program Optimization in Deep Neural Network Compilation

Yoloxkeypointsegment - An anchor-free version of YOLO, with a simpler design but better performance

HCQ: Hybrid Contrastive Quantization for Efficient Cross-View Video Retrieval

Code repository for the paper Computer Vision User Entity Behavior Analytics

Deep Federated Learning for Autonomous Driving

Code from PropMix, accepted at BMVC'21

Kaggle-titanic - A tutorial for Kaggle's Titanic: Machine Learning from Disaster competition. Demonstrates basic data munging, analysis, and visualization techniques. Shows examples of supervised machine learning techniques.

Train the HRNet model on ImageNet

Repository of Jupyter notebook tutorials for teaching the Deep Learning Course at the University of Amsterdam (MSc AI), Fall 2020

AISTATS 2019: Confidence-based Graph Convolutional Networks for Semi-Supervised Learning

Negative Sample Matters: A Renaissance of Metric Learning for Temporal Grounding

Two-Stream Adaptive Graph Convolutional Networks for Skeleton-Based Action Recognition in CVPR19

Spatially-Adaptive Pixelwise Networks for Fast Image Translation, CVPR 2021

Annotate datasets with a semi-trained or fully trained YOLOv5 model

WORD: Revisiting Organs Segmentation in the Whole Abdominal Region

simple artificial intelligence utilities

This repository is an implementation of our NeurIPS 2021 paper (Stylized Dialogue Generation with Multi-Pass Dual Learning) in PyTorch.

This is code of book "Learn Deep Learning with PyTorch"