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

Torch implementation of various types of GAN (e.g. DCGAN, ALI, Context-encoder, DiscoGAN, CycleGAN, EBGAN, LSGAN)

In this project, two programs can help you take full agvantage of time on the model training with a remote server

An original implementation of "MetaICL Learning to Learn In Context" by Sewon Min, Mike Lewis, Luke Zettlemoyer and Hannaneh Hajishirzi

An implementation of MobileFormer

Learning Continuous Image Representation with Local Implicit Image Function

OpenLT: An open-source project for long-tail classification

Learning with Noisy Labels via Sparse Regularization, ICCV2021

AnimationKit: AI Upscaling & Interpolation using Real-ESRGAN+RIFE

PyTorchCV: A PyTorch-Based Framework for Deep Learning in Computer Vision.

使用yolov5训练自己数据集(详细过程)并通过flask部署

OCR Streamlit App is used to extract text from images using python's easyocr, pytorch and streamlit packages

Architecture Patterns with Python (TDD, DDD, EDM)

Rotation-Only Bundle Adjustment

Code for "Single-view robot pose and joint angle estimation via render & compare", CVPR 2021 (Oral).

Pytorch implementation for the paper: Contrastive Learning for Cold-start Recommendation

Custom IMDB Dataset is extracted between 2020-2021 and custom distilBERT model is trained for movie success probability prediction

Character-Input - Create a program that asks the user to enter their name and their age

The final project of "Applying AI to 3D Medical Imaging Data" from "AI for Healthcare" nanodegree - Udacity.

PyTorch implementation of Spiking Neural Networks trained on surrogate gradient & BPTT using snntorch.

CapsuleVOS: Semi-Supervised Video Object Segmentation Using Capsule Routing