Skip to content

Commit 8d53be1

Browse files
salvacortssalvacorts
committed
Notebook
1 parent 9675489 commit 8d53be1

File tree

10 files changed

+1153
-3
lines changed

10 files changed

+1153
-3
lines changed

‎Dockerfile‎

Lines changed: 39 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,41 @@
1-
FROM nvcr.io/nvidia/dli/dli-nano-ai:v2.0.1-r32.5.0
1+
FROM nvcr.io/nvidia/l4t-ml:r32.5.0-py3
2+
3+
# Install basic dependencies
4+
RUN apt update
5+
RUN apt install -y cmake
6+
RUN apt install -y libprotobuf-dev protobuf-compiler
27

38
# Intall detectron2
4-
!pip3 install 'git+https://github.com/facebookresearch/detectron2.git'
9+
RUN pip3 install pyyaml==5.1
10+
RUN pip3 install 'git+https://github.com/facebookresearch/detectron2.git'
11+
12+
# Install onnx and tensorrt backend
13+
RUN pip3 install onnx==1.6.0
14+
RUN pip3 install 'git+https://github.com/onnx/onnx-tensorrt.git'
15+
RUN git clone https://github.com/NVIDIA-AI-IOT/torch2trt
16+
RUN cd torch2trt && python3 setup.py install --plugins
17+
18+
# Install jupyterlab
19+
RUN apt install -y curl libffi-dev
20+
RUN curl -sL https://deb.nodesource.com/setup_12.x | bash -
21+
RUN apt install -y nodejs
22+
RUN pip3 install --upgrade jupyterlab
23+
RUN jupyter lab build
24+
25+
# Install jetcam
26+
RUN pip3 install 'git+https://github.com/NVIDIA-AI-IOT/jetcam.git'
27+
28+
# Install jupyter widgets
29+
RUN pip3 install ipywidgets
30+
RUN jupyter nbextension enable --py widgetsnbextension
31+
RUN jupyter labextension install @jupyter-widgets/jupyterlab-manager
32+
33+
# Paths to mount data
34+
VOLUME [/usr/lib/python3.6/dist-packages/tensorrt]
35+
VOLUME [/usr/src/tensorrt]
36+
VOLUME [/tmp/argus_socket]
37+
VOLUME [/tmp/.X11-unix/]
38+
VOLUME [/data]
39+
40+
WORKDIR /data
41+
#ENTRYPOINT jupyter lab --allow-root --config /jupyter_conf.py

‎README.md‎

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,9 @@
44

55
## Training
66

7-
## Inference
7+
## Inference
8+
9+
### Build Dockerfile
10+
```bash
11+
sudo docker build -t torch_detectron .
12+
```

‎inference.ipynb‎

Lines changed: 1092 additions & 0 deletions
Large diffs are not rendered by default.

‎jupyter_conf.py‎

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
c.NotebookApp.ip = '*'
2+
c.NotebookApp.token = "1234"
3+
c.NotebookApp.open_browser = False
4+
c.NotebookApp.port = 8888

‎mstd_metadata.json‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
{"name": "mtsd_train", "thing_classes": ["regulatory--end-of-bicycles-only--g1", "regulatory--no-parking-or-no-stopping--g2", "regulatory--no-motor-vehicles--g4", "information--bus-stop--g1", "warning--traffic-merges-right--g2", "complementary--distance--g1", "regulatory--go-straight-or-turn-left--g2", "regulatory--no-overtaking--g5", "warning--divided-highway-ends--g1", "regulatory--no-pedestrians-or-bicycles--g1", "regulatory--no-parking--g1", "information--emergency-facility--g2", "regulatory--bicycles-only--g3", "warning--horizontal-alignment-right--g3", "warning--two-way-traffic--g1", "information--disabled-persons--g1", "warning--offset-roads--g3", "warning--texts--g2", "information--food--g2", "warning--t-roads--g2", "regulatory--end-of-priority-road--g1", "regulatory--no-vehicles-carrying-dangerous-goods--g1", "information--pedestrians-crossing--g1", "regulatory--mopeds-and-bicycles-only--g1", "regulatory--road-closed-to-vehicles--g1", "warning--texts--g1", "regulatory--maximum-speed-limit-led-100--g1", "regulatory--parking-restrictions--g2", "complementary--go-left--g1", "regulatory--bicycles-only--g2", "regulatory--detour-left--g1", "complementary--keep-left--g1", "information--airport--g1", "regulatory--go-straight-or-turn-right--g1", "warning--traffic-merges-left--g1", "regulatory--passing-lane-ahead--g1", "warning--pedestrians-crossing--g5", "warning--traffic-merges-right--g1", "warning--junction-with-a-side-road-perpendicular-right--g1", "regulatory--reversible-lanes--g2", "complementary--tow-away-zone--g1", "regulatory--maximum-speed-limit-30--g3", "regulatory--pedestrians-only--g1", "complementary--chevron-right--g1", "warning--trail-crossing--g2", "complementary--pass-right--g1", "warning--curve-left--g1", "regulatory--no-parking-or-no-stopping--g1", "warning--traffic-signals--g1", "regulatory--maximum-speed-limit-40--g3", "regulatory--yield--g1", "regulatory--no-motor-vehicles-except-motorcycles--g1", "information--end-of-built-up-area--g1", "regulatory--maximum-speed-limit-50--g6", "warning--hairpin-curve-left--g1", "regulatory--no-turn-on-red--g3", "regulatory--end-of-maximum-speed-limit-30--g2", "regulatory--maximum-speed-limit-40--g6", "warning--uneven-road--g6", "regulatory--no-stopping--g4", "information--limited-access-road--g1", "regulatory--turn-right--g2", "warning--domestic-animals--g3", "warning--railroad-crossing-with-barriers--g4", "regulatory--turning-vehicles-yield-to-pedestrians--g1", "regulatory--dual-lanes-turn-left-no-u-turn--g1", "warning--narrow-bridge--g1", "regulatory--maximum-speed-limit-30--g1", "regulatory--maximum-speed-limit-100--g1", "regulatory--one-way-right--g1", "warning--roundabout--g1", "information--parking--g5", "regulatory--no-pedestrians--g2", "regulatory--do-not-block-intersection--g1", "complementary--distance--g2", "regulatory--no-mopeds-or-bicycles--g1", "warning--falling-rocks-or-debris-right--g4", "warning--crossroads--g3", "information--tram-bus-stop--g2", "regulatory--no-stopping--g2", "regulatory--maximum-speed-limit-15--g1", "warning--two-way-traffic--g2", "regulatory--maximum-speed-limit-25--g1", "regulatory--minimum-safe-distance--g1", "warning--other-danger--g1", "regulatory--shared-path-bicycles-and-pedestrians--g1", "regulatory--no-left-turn--g3", "regulatory--maximum-speed-limit-90--g1", "complementary--maximum-speed-limit-40--g1", "warning--pedestrians-crossing--g12", "warning--double-curve-first-right--g1", "regulatory--no-u-turn--g1", "regulatory--pass-on-either-side--g2", "regulatory--turn-left--g3", "warning--railroad-intersection--g3", "regulatory--u-turn--g1", "regulatory--end-of-speed-limit-zone--g1", "regulatory--end-of-prohibition--g1", "warning--wild-animals--g4", "regulatory--stop-here-on-red-or-flashing-light--g1", "warning--slippery-road-surface--g1", "regulatory--lane-control--g1", "warning--traffic-signals--g3", "regulatory--maximum-speed-limit-20--g1", "complementary--accident-area--g3", "regulatory--stop-signals--g1", "warning--hairpin-curve-left--g3", "regulatory--truck-speed-limit-60--g1", "warning--road-widens--g1", "regulatory--no-right-turn--g3", "regulatory--no-motorcycles--g1", "regulatory--no-straight-through--g2", "regulatory--dual-path-pedestrians-and-bicycles--g1", "regulatory--end-of-maximum-speed-limit-70--g1", "warning--road-bump--g1", "warning--slippery-road-surface--g2", "regulatory--one-way-left--g2", "complementary--go-right--g1", "information--end-of-living-street--g1", "regulatory--no-hawkers--g1", "regulatory--no-overtaking--g2", "regulatory--keep-right--g2", "regulatory--no-parking-or-no-stopping--g3", "warning--other-danger--g3", "complementary--distance--g3", "warning--school-zone--g2", "warning--railroad-crossing-with-barriers--g2", "warning--pedestrians-crossing--g10", "regulatory--text-four-lines--g1", "regulatory--turn-right--g1", "warning--railroad-crossing--g1", "information--dead-end--g1", "warning--double-curve-first-left--g2", "warning--winding-road-first-left--g1", "warning--pedestrian-stumble-train--g1", "regulatory--no-u-turn--g2", "regulatory--maximum-speed-limit-5--g1", "warning--curve-right--g1", "complementary--trucks-turn-right--g1", "warning--accidental-area-unsure--g2", "complementary--both-directions--g1", "regulatory--no-motor-vehicles--g1", "information--end-of-motorway--g1", "regulatory--go-straight--g1", "complementary--chevron-left--g5", "warning--railroad-crossing-without-barriers--g4", "information--parking--g1", "complementary--turn-left--g2", "warning--kangaloo-crossing--g1", "warning--equestrians-crossing--g2", "regulatory--roundabout--g1", "warning--junction-with-a-side-road-acute-left--g1", "warning--turn-left--g1", "information--end-of-pedestrians-only--g2", "regulatory--no-right-turn--g1", "warning--falling-rocks-or-debris-right--g1", "regulatory--maximum-speed-limit-80--g1", "warning--railroad-crossing-without-barriers--g3", "regulatory--weight-limit-with-trucks--g1", "complementary--priority-route-at-intersection--g1", "warning--curve-right--g2", "warning--height-restriction--g2", "information--safety-area--g2", "information--highway-interstate-route--g2", "information--dead-end-except-bicycles--g1", "information--highway-exit--g1", "regulatory--turn-right-ahead--g2", "warning--stop-ahead--g9", "warning--railroad-crossing--g3", "warning--flaggers-in-road--g1", "warning--double-turn-first-right--g1", "regulatory--triple-lanes-turn-left-center-lane--g1", "warning--curve-left--g2", "regulatory--stop-here-on-red-or-flashing-light--g2", "warning--bicycles-crossing--g3", "complementary--maximum-speed-limit-50--g1", "warning--winding-road-first-right--g3", "regulatory--one-way-left--g3", "warning--wild-animals--g1", "regulatory--no-heavy-goods-vehicles--g4", "regulatory--no-entry--g1", "warning--wombat-crossing--g1", "warning--emergency-vehicles--g1", "warning--roundabout--g25", "regulatory--give-way-to-oncoming-traffic--g1", "warning--pedestrians-crossing--g1", "regulatory--stop--g1", "complementary--keep-right--g1", "regulatory--keep-left--g2", "warning--pedestrians-crossing--g11", "information--children--g1", "regulatory--no-buses--g3", "warning--railroad-intersection--g4", "regulatory--no-u-turn--g3", "information--parking--g45", "warning--road-widens-right--g1", "regulatory--road-closed-to-vehicles--g3", "information--end-of-limited-access-road--g1", "information--pedestrians-crossing--g2", "warning--steep-ascent--g7", "regulatory--pass-on-either-side--g1", "regulatory--wrong-way--g1", "regulatory--bicycles-only--g1", "regulatory--buses-only--g1", "warning--falling-rocks-or-debris-right--g2", "warning--trucks-crossing--g1", "regulatory--one-way-straight--g1", "warning--roadworks--g2", "regulatory--no-pedestrians--g3", "information--telephone--g2", "regulatory--go-straight-or-turn-right--g3", "regulatory--no-overtaking--g4", "warning--divided-highway-ends--g2", "complementary--one-direction-left--g1", "warning--road-narrows-left--g1", "regulatory--do-not-stop-on-tracks--g1", "regulatory--weight-limit--g1", "warning--pass-left-or-right--g2", "regulatory--one-way-right--g3", "regulatory--turn-right-ahead--g1", "regulatory--road-closed--g2", "complementary--except-bicycles--g1", "warning--narrow-bridge--g3", "regulatory--maximum-speed-limit-40--g1", "information--gas-station--g3", "regulatory--no-motor-vehicle-trailers--g1", "warning--pedestrians-crossing--g4", "warning--junction-with-a-side-road-perpendicular-left--g3", "regulatory--keep-right--g6", "regulatory--roundabout--g2", "regulatory--maximum-speed-limit-120--g1", "warning--turn-right--g2", "warning--road-narrows--g1", "regulatory--maximum-speed-limit-led-80--g1", "warning--restricted-zone--g1", "warning--road-narrows-left--g2", "information--stairs--g1", "warning--uneven-roads-ahead--g1", "information--living-street--g1", "regulatory--turn-left--g1", "complementary--maximum-speed-limit-55--g1", "complementary--chevron-right--g4", "regulatory--maximum-speed-limit-35--g2", "information--parking--g3", "warning--bicycles-crossing--g1", "warning--playground--g1", "regulatory--no-heavy-goods-vehicles--g2", "complementary--maximum-speed-limit-15--g1", "regulatory--dual-lanes-turn-left-or-straight--g1", "information--camp--g1", "regulatory--turn-left-ahead--g1", "regulatory--no-motor-vehicles-except-motorcycles--g2", "warning--slippery-motorcycles--g1", "information--lodging--g1", "regulatory--dual-lanes-go-straight-on-right--g1", "warning--loop-270-degree--g1", "regulatory--keep-right--g4", "regulatory--maximum-speed-limit-led-60--g1", "warning--trams-crossing--g1", "complementary--chevron-left--g3", "regulatory--go-straight-or-turn-left--g1", "warning--hairpin-curve-right--g1", "regulatory--no-turns--g1", "warning--road-narrows-right--g2", "complementary--chevron-right-unsure--g6", "warning--horizontal-alignment-right--g1", "warning--t-roads--g1", "regulatory--no-overtaking-by-heavy-goods-vehicles--g1", "regulatory--no-bicycles--g3", "regulatory--no-straight-through--g1", "warning--roadworks--g3", "complementary--maximum-speed-limit-20--g1", "regulatory--maximum-speed-limit-55--g2", "complementary--maximum-speed-limit-35--g1", "warning--junction-with-a-side-road-acute-right--g1", "information--trailer-camping--g1", "regulatory--no-left-turn--g2", "warning--domestic-animals--g1", "regulatory--priority-over-oncoming-vehicles--g1", "regulatory--maximum-speed-limit-110--g1", "other-sign", "warning--double-curve-first-right--g2", "warning--double-curve-first-left--g1", "warning--junction-with-a-side-road-perpendicular-left--g1", "warning--pedestrians-crossing--g9", "information--parking--g2", "complementary--buses--g1", "complementary--maximum-speed-limit-25--g1", "complementary--chevron-left--g4", "warning--junction-with-a-side-road-perpendicular-right--g3", "regulatory--pedestrians-only--g2", "regulatory--stop--g10", "information--gas-station--g1", "complementary--maximum-speed-limit-30--g1", "regulatory--turn-left--g2", "information--hospital--g1", "warning--railroad-crossing-without-barriers--g1", "complementary--turn-right--g2", "warning--dip--g2", "regulatory--no-heavy-goods-vehicles--g5", "regulatory--no-parking--g5", "warning--y-roads--g1", "information--bike-route--g1", "information--central-lane--g1", "regulatory--maximum-speed-limit-50--g1", "regulatory--no-left-turn--g1", "regulatory--dual-lanes-turn-right-or-straight--g1", "regulatory--radar-enforced--g1", "information--parking--g6", "warning--crossroads--g1", "regulatory--no-right-turn--g2", "regulatory--no-parking--g2", "warning--bicycles-crossing--g2", "warning--roadworks--g4", "regulatory--end-of-no-parking--g1", "information--airport--g2", "information--telephone--g1", "warning--uneven-road--g2", "regulatory--no-overtaking--g1", "regulatory--turn-right--g3", "warning--playground--g3", "complementary--obstacle-delineator--g2", "regulatory--priority-road--g4", "information--interstate-route--g1", "warning--road-bump--g2", "regulatory--maximum-speed-limit-70--g1", "regulatory--no-stopping--g8", "warning--turn-right--g1", "warning--traffic-signals--g4", "regulatory--maximum-speed-limit-100--g3", "warning--junction-with-a-side-road-perpendicular-left--g4", "regulatory--no-heavy-goods-vehicles--g1", "regulatory--no-turn-on-red--g2", "warning--hairpin-curve-right--g4", "regulatory--end-of-buses-only--g1", "warning--shared-lane-motorcycles-bicycles--g1", "warning--winding-road-first-right--g1", "regulatory--maximum-speed-limit-45--g3", "regulatory--no-heavy-goods-vehicles-or-buses--g1", "regulatory--stop--g2", "complementary--maximum-speed-limit-70--g1", "complementary--go-right--g2", "regulatory--maximum-speed-limit-60--g1", "complementary--obstacle-delineator--g1", "regulatory--width-limit--g1", "information--motorway--g1", "regulatory--left-turn-yield-on-green--g1", "regulatory--road-closed--g1", "warning--dual-lanes-right-turn-or-go-straight--g1", "warning--railroad-crossing-with-barriers--g1", "warning--road-narrows-right--g1", "regulatory--maximum-speed-limit-10--g1", "complementary--maximum-speed-limit-75--g1", "complementary--trucks--g1", "complementary--maximum-speed-limit-45--g1", "regulatory--no-motorcycles--g2", "regulatory--dual-lanes-go-straight-on-left--g1", "information--no-parking--g3", "complementary--chevron-right--g5", "warning--road-narrows--g2", "regulatory--no-bicycles--g1", "complementary--chevron-right--g3", "regulatory--maximum-speed-limit-25--g2", "warning--horizontal-alignment-left--g1", "complementary--chevron-left--g1", "regulatory--no-turn-on-red--g1", "regulatory--go-straight--g3", "complementary--one-direction-right--g1", "warning--texts--g3", "regulatory--one-way-left--g1", "complementary--extent-of-prohibition-area-both-direction--g1", "warning--added-lane-right--g1", "complementary--chevron-left--g2", "warning--traffic-signals--g2", "regulatory--dual-path-bicycles-and-pedestrians--g1", "warning--children--g2", "warning--children--g1", "regulatory--height-limit--g1", "regulatory--keep-left--g1", "warning--crossroads-with-priority-to-the-right--g1", "warning--roadworks--g6", "information--road-bump--g1", "regulatory--no-stopping--g5", "information--minimum-speed-40--g1", "regulatory--end-of-maximum-speed-limit-70--g2", "warning--double-reverse-curve-right--g1", "regulatory--shared-path-pedestrians-and-bicycles--g1", "regulatory--no-stopping--g15", "regulatory--one-way-right--g2", "warning--railroad-crossing--g4", "regulatory--maximum-speed-limit-65--g2", "warning--bus-stop-ahead--g3", "regulatory--go-straight-or-turn-left--g3", "regulatory--maximum-speed-limit-45--g1", "warning--pass-left-or-right--g1", "regulatory--no-pedestrians--g1", "warning--winding-road-first-left--g2", "warning--traffic-merges-left--g2", "regulatory--keep-right--g1", "regulatory--no-bicycles--g2", "warning--roadworks--g1"]}

0 commit comments

Comments
 (0)