Code Monkey home page Code Monkey logo

cts-uda's Introduction

CTS Sim-to-Real

Installation

Follow instruction in OpenPCDet Install Guide

Get Started

Dataset Preparation

Carla3D (Simulation/Source) is available at not released yet.

TinySUSCape (already organized in KITTI format) is available from JST.

Please refer to 3D_adapt_auto_driving to prepare the KITTI and Lyft dataset. The train/val split txt of KITTI and Lyft used in our experiments is same as the 3D_adapt_auto_driving.

Build dataset info db with:

cd tools/
python -m pcdet.datasets.kitti.kitti_dataset create_kitti_infos cfgs/dataset_configs/{DATASET}.yaml

Training

  1. Train a source model on simulation domain (Carla3D):
cd tools/
python train.py --cfg_file cfgs/carla_models/pointrcnn_carla_org.yaml
  1. Self-training with target domain dataset:
python train_st.py 
--cfg_file cfgs/carla_models/pointrcnn_org_to_{DATASET}.yaml \
--mining_at 0 10 30 40 --mining_portion 0.3 0.5 0.7 1.0 \
# for lyft
# --mining_at 0 5 15 25 --mining_portion 0.3 0.5 0.7 1.0 \
--pretrained_model {BEST_MODEL_FROM_SOURCE_TRAIN} \

Testing

python test.py \ 
--cfg_file cfgs/carla_models/pointrcnn_org_to_{DATASET}.yaml \
--batch_size 8 --workers 8 \
--ckpt {PATH_TO_CHECKPOINT}

Acknowledgement

Our code is based on OpenPCDet and JST

cts-uda's People

Contributors

tendo518 avatar

Watchers

 avatar

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.