The official repository for paper ''Domain Generalization for Vision-based Driving Trajectory Generation'' submitted to ICRA 2022

Last update: Sep 26, 2022

Related tags

Deep Learning DG-TrajGen

Overview

DG-TrajGen

The official repository for paper ''Domain Generalization for Vision-based Driving Trajectory Generation'' submitted to ICRA 2022.

Our Method

Trajectory representation:
- Model: ./learning/model.py/Generator
Latent Action Space Learning:
- Generator model: ./learning/model.py/Generator
- Discriminator model: ./learning/model.py/Discriminator
- Training: ./scripts/Ours/stage1_train_GAN.py
Encoder Pre-training:
- Training: ./scripts/Ours/stage2_pretrain_encoder.py
End-to-End Training the Encoder:
- Training: ./scripts/Ours/stage3_train_e2e.py

Comparative Study

RIP:
- Training: ./scripts/RIP/train.py
- Referenced official code: github
- Paper: arxiv
MixStyle:
- Training: ./scripts/MixStyle/train.py
- Referenced official code: github
- Paper: arxiv
DIVA:
- Training: ./scripts/DIVA/train.py
- Referenced official code: github
- Paper: arxiv
DAL:
- Training: ./scripts/DAL/train.py
E2E NT:
- Training: ./scripts/E2ENT/train.py
- Referenced official code: github
- Paper: arxiv

Closed-loop Experiments:

We train the model on the Oxford RobotCar dataset and directly generalize it to the CARLA simulation.

Run: ./scripts/CARLA/run_ours.py

Citation

If you use our source code, please consider citing the following:

@article{wang2021domain,
  title={Domain Generalization for Vision-based Driving Trajectory Generation},
  author={Wang, Yunkai and Zhang, Dongkun and Cui, Yuxiang and Chen, Zexi and Jing, Wei and Chen, Junbo and Xiong, Rong and Wang, Yue},
  journal={arXiv preprint arXiv:2109.13858},
  year={2021}
}

Comments

no module named 'rllib.basic'

Hello, when I try to run stage1_train_GAN.py, there will be an error: "no module named 'rllib.basic' ", when I try to check the error, I found that it is in the system folder under Carla_utils There was an error in init.py. At first, when I tried to pip install, I would get error: no matching distribution found for rllib.basic. Is this a library or something? Thanks again for being able to see my question, looking forward to your answer

opened by hickey23 0
Some files are missing

Thank you for sharing your code, but it seems that some files such as "carla_utils" and "robo_utils" and the corresponding pth files are missing, which makes it impossible to train or run "run_ours.py"......

opened by JSLOUJSKOU 2

Official implementation of paper Gradient Matching for Domain Generalization

Gradient Matching for Domain Generalisation This is the official PyTorch implementation of Gradient Matching for Domain Generalisation. In our paper,

94 Dec 23, 2022

Official implementation of "Learning Forward Dynamics Model and Informed Trajectory Sampler for Safe Quadruped Navigation" (RSS 2022)

Intro Official implementation of "Learning Forward Dynamics Model and Informed Trajectory Sampler for Safe Quadruped Navigation" Robotics:Science and

21 Dec 7, 2022

Offcial repository for the IEEE ICRA 2021 paper Auto-Tuned Sim-to-Real Transfer.

47 Jun 30, 2022

SafePicking: Learning Safe Object Extraction via Object-Level Mapping, ICRA 2022

SafePicking Learning Safe Object Extraction via Object-Level Mapping Kentaro Wad

49 Oct 24, 2022

[ICRA 2022] CaTGrasp: Learning Category-Level Task-Relevant Grasping in Clutter from Simulation

This is the official implementation of our paper: Bowen Wen, Wenzhao Lian, Kostas Bekris, and Stefan Schaal. "CaTGrasp: Learning Category-Level Task-R

199 Jan 4, 2023

the official code for ICRA 2021 Paper: "Multimodal Scale Consistency and Awareness for Monocular Self-Supervised Depth Estimation"

G2S This is the official code for ICRA 2021 Paper: Multimodal Scale Consistency and Awareness for Monocular Self-Supervised Depth Estimation by Hemang

4 Jul 27, 2022

Official PyTorch implementation of the ICRA 2021 paper: Adversarial Differentiable Data Augmentation for Autonomous Systems.

Adversarial Differentiable Data Augmentation This repository provides the official PyTorch implementation of the ICRA 2021 paper: Adversarial Differen

3 Oct 15, 2022

This is the official code of our paper "Diversity-based Trajectory and Goal Selection with Hindsight Experience Relay" (PRICAI 2021)

Diversity-based Trajectory and Goal Selection with Hindsight Experience Replay This is the official implementation of our paper "Diversity-based Traje

6 Jul 18, 2022

Official repository of the AAAI'2022 paper "Contrast and Generation Make BART a Good Dialogue Emotion Recognizer"

CoG-BART Contrast and Generation Make BART a Good Dialogue Emotion Recognizer Quick Start: To run the model on test sets of four datasets, Download th

39 Dec 24, 2022

model(Dec 14, 2021)

64 dim, trained on the Oxford RobotCar dataset
Source code(tar.gz)
Source code(zip)
generator.pth(2.63 MB)
dataset(Dec 14, 2021)

Extract to ./robo_utils/oxford/
Source code(tar.gz)
Source code(zip)
models.zip(67.46 MB)

The official repository for paper ''Domain Generalization for Vision-based Driving Trajectory Generation'' submitted to ICRA 2022

Related tags

Overview

DG-TrajGen

Our Method

Comparative Study

Closed-loop Experiments:

Citation

You might also like...

Official implementation of paper Gradient Matching for Domain Generalization

Official implementation of "Learning Forward Dynamics Model and Informed Trajectory Sampler for Safe Quadruped Navigation" (RSS 2022)

Offcial repository for the IEEE ICRA 2021 paper Auto-Tuned Sim-to-Real Transfer.

SafePicking: Learning Safe Object Extraction via Object-Level Mapping, ICRA 2022

[ICRA 2022] CaTGrasp: Learning Category-Level Task-Relevant Grasping in Clutter from Simulation

the official code for ICRA 2021 Paper: "Multimodal Scale Consistency and Awareness for Monocular Self-Supervised Depth Estimation"

Official PyTorch implementation of the ICRA 2021 paper: Adversarial Differentiable Data Augmentation for Autonomous Systems.

This is the official code of our paper "Diversity-based Trajectory and Goal Selection with Hindsight Experience Relay" (PRICAI 2021)

Official repository of the AAAI'2022 paper "Contrast and Generation Make BART a Good Dialogue Emotion Recognizer"

Comments

no module named 'rllib.basic'

Some files are missing

Releases(model)

model(Dec 14, 2021)

dataset(Dec 14, 2021)

Owner

Wang

This is the codebase for the ICLR 2021 paper Trajectory Prediction using Equivariant Continuous Convolution

AFL binary instrumentation

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.

Final term project for Bayesian Machine Learning Lecture (XAI-623)

[ICLR 2021] Heteroskedastic and Imbalanced Deep Learning with Adaptive Regularization

Demo code for paper "Learning optical flow from still images", CVPR 2021.

A Pytorch Implementation of a continuously rate adjustable learned image compression framework.

TResNet: High Performance GPU-Dedicated Architecture

The codes and related files to reproduce the results for Image Similarity Challenge Track 1.

Clockwork Variational Autoencoder

Recurrent Neural Network Tutorial, Part 2 - Implementing a RNN in Python and Theano

Implements Stacked-RNN in numpy and torch with manual forward and backward functions

E2EC: An End-to-End Contour-based Method for High-Quality High-Speed Instance Segmentation

[IEEE TPAMI21] MobileSal: Extremely Efficient RGB-D Salient Object Detection [PyTorch & Jittor]

A clean and extensible PyTorch implementation of Masked Autoencoders Are Scalable Vision Learners

Unofficial PyTorch implementation of MobileViT.

Not Suitable for Work (NSFW) classification using deep neural network Caffe models.

Code release for "Making a Bird AI Expert Work for You and Me".

The codes and models in 'Gaze Estimation using Transformer'.

Code for "Diversity can be Transferred: Output Diversification for White- and Black-box Attacks"