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

Implemented fully documented Particle Swarm Optimization algorithm (basic model with few advanced features) using Python programming language

Trajectory Extraction of road users via Traffic Camera

Use evolutionary algorithms instead of gridsearch in scikit-learn

PyTorch code for the paper "Complementarity is the King: Multi-modal and Multi-grained Hierarchical Semantic Enhancement Network for Cross-modal Retrieval".

Convert onnx models to pytorch.

Deep Text Search is an AI-powered multilingual text search and recommendation engine with state-of-the-art transformer-based multilingual text embedding (50+ languages).

Reverse engineering recurrent neural networks with Jacobian switching linear dynamical systems

Py-FEAT: Python Facial Expression Analysis Toolbox

LLVM-based compiler for LightGBM gradient-boosted trees. Speeds up prediction by ≥10x.

Implementation of Enformer, Deepmind's attention network for predicting gene expression, in Pytorch

Backend code to use MCPI's python API to make infinite worlds with custom generation

YolactEdge: Real-time Instance Segmentation on the Edge

[ECE NTUA] 👁 Computer Vision - Lab Projects & Theoretical Problem Sets (2020-2021)

Code for KDD'20 "An Efficient Neighborhood-based Interaction Model for Recommendation on Heterogeneous Graph"

An e-commerce company wants to segment its customers and determine marketing strategies according to these segments.

Self-labelling via simultaneous clustering and representation learning. (ICLR 2020)

VolumeGAN - 3D-aware Image Synthesis via Learning Structural and Textural Representations

A Haskell kernel for IPython.

TVNet: Temporal Voting Network for Action Localization

The FIRST GANs-based omics-to-omics translation framework