Code Monkey home page Code Monkey logo

horizon_oe's Introduction

horizon_oe

Quickstart

0. Setting Horizon Env.

$bash run_docker.sh /data gpu

1. Prepare Dataset

pack nuscenes dataset

# To pack train set
$python3 ./tools/nuscenes_packer.py --src-data-dir ./data/nuscenes/ --pack-type lmdb --target-data-dir ./tmp_data/nuscenes/v1.0-trainval --version v1.0-trainval --split-name train
# To pack val set
$python3 ./tools/nuscenes_packer.py --src-data-dir ./data/nuscenes/ --pack-type lmdb --target-data-dir ./tmp_data/nuscenes/v1.0-trainval --version v1.0-trainval --split-name val

The tree strcture of nuscenes packed is as following:

|-- tmp_data 
|   |-- nuscenes 
|   |   |-- v1.0-trainval
|   |   |   |-- train_lmdb  #packed train set
|   |   |   |   |-- data.mdb
|   |   |   |   `-- lock.mdb
|   |   |   `-- val_lmdb   #packed val set
|   |   |   |   |-- data.mdb
|   |   |   |   `-- lock.mdb

Decompress v1.0-trainval_meta.tar and nuscenes-map-extension-v1.3.zip, then reconstruct as following:

|-- tmp_data 
|   |-- nuscenes 
|   |   |-- meta
|   |   |   |-- maps  
|   |   |   |   |-- 36092f0b03a857c6a3403e25b4b7aab3.png
|   |   |   |   |-- ...
|   |   |   |   |-- 93406b464a165eaba6d9de76ca09f5da.png
|   |   |   |   |-- prediction
|   |   |   |   |-- basemap
|   |   |   |   |-- expansion
|   |   |   |-- v1.0-trainval  
|   |   |       |-- attribute.json
|   |   |           ...
|   |   |       |-- visibility.json
|   |   `-- v1.0-trainval 
|   |   |   |-- train_lmdb  
|   |   |   `-- val_lmdb   

2. Train

modify configs according to your device

$cd /open_explorer/worksapce/
$python3 tools/train.py --config configs/bev_ipm_efficientnetb0_multitask_nuscenes.py --stage float

3. Evaluation

$python3 tools/predic.py --config configs/bev_ipm_efficientnetb0_multitask_nuscenes.py --stage float

4. Infer

python3 tools/infer.py --config configs/bev/bev_mt_ipm_efficientnetb0_nuscenes.py --model-inputs imagedir:${imagedir},homo:${homography.npy}

horizon_oe's People

Contributors

hadonga avatar donghe4 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.