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

Probabilistic Entity Representation Model for Reasoning over Knowledge Graphs

Learning based AI for playing multi-round Koi-Koi hanafuda card games. Have fun.

[PyTorch] Official implementation of CVPR2021 paper "PointDSC: Robust Point Cloud Registration using Deep Spatial Consistency". https://arxiv.org/abs/2103.05465

An 16kHz implementation of HiFi-GAN for soft-vc.

a general-purpose Transformer based vision backbone

the code for paper "Energy-Based Open-World Uncertainty Modeling for Confidence Calibration"

Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms

Scheduling BilinearRewards

Official Pytorch implementation of the paper "Action-Conditioned 3D Human Motion Synthesis with Transformer VAE", ICCV 2021

Retrieve and analysis data from SDSS (Sloan Digital Sky Survey)

Project for music generation system based on object tracking and CGAN

利用python脚本实现微信、支付宝账单的合并，并保存到excel文件实现自动记账，可查看可视化图表。

Hl classification bc - A Network-Based High-Level Data Classification Algorithm Using Betweenness Centrality

Goal of the project : Detecting Temporal Boundaries in Sign Language videos

Forest R-CNN: Large-Vocabulary Long-Tailed Object Detection and Instance Segmentation (ACM MM 2020)

NExT-QA: Next Phase of Question-Answering to Explaining Temporal Actions (CVPR2021)

TF2 implementation of knowledge distillation using the "function matching" hypothesis from the paper Knowledge distillation: A good teacher is patient and consistent by Beyer et al.

Development kit for MIT Scene Parsing Benchmark

AutoML library for deep learning

Official implementation of NeurIPS 2021 paper "Contextual Similarity Aggregation with Self-attention for Visual Re-ranking"