Skip to content

Commit

Permalink
fix: update edge_auto_launcher tasks (#57)
Browse files Browse the repository at this point in the history
  • Loading branch information
tokuda99 authored Sep 5, 2024
1 parent d33b3d3 commit 53c50af
Show file tree
Hide file tree
Showing 8 changed files with 107 additions and 43 deletions.
5 changes: 4 additions & 1 deletion .cspell.json
Original file line number Diff line number Diff line change
Expand Up @@ -152,6 +152,9 @@
"soruce",
"TIERIV",
"txqueue",
"txqueuelen"
"txqueuelen",
"semseg",
"adlinktech",
"covdep"
]
}
1 change: 1 addition & 0 deletions ansible/roles/artifacts/defaults/main.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
artifacts_script_download_dir: /tmp
78 changes: 78 additions & 0 deletions ansible/roles/artifacts/tasks/main.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,78 @@
# tensorrt_yolox
- name: Create tensorrt_yolox directory inside {{ data_dir }}
ansible.builtin.file:
path: "{{ data_dir }}/tensorrt_yolox"
mode: "755"
state: directory

- name: Download tensorrt_yolox/yolox-tiny.onnx
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/yolox-tiny.onnx
dest: "{{ data_dir }}/tensorrt_yolox/yolox-tiny.onnx"
mode: "644"
checksum: sha256:471a665f4243e654dff62578394e508db22ee29fe65d9e389dfc3b0f2dee1255

- name: Download tensorrt_yolox/yolox-sPlus-opt.onnx
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/yolox-sPlus-opt.onnx
dest: "{{ data_dir }}/tensorrt_yolox/yolox-sPlus-opt.onnx"
mode: "644"
checksum: sha256:36b0832177b01e6b278e00c7369f1de71e616c36261cbae50f0753d41289da01

- name: Download tensorrt_yolox/yolox-sPlus-opt.EntropyV2-calibration.table
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/yolox-sPlus-opt.EntropyV2-calibration.table
dest: "{{ data_dir }}/tensorrt_yolox/yolox-sPlus-opt.EntropyV2-calibration.table"
mode: "644"
checksum: sha256:b9e9d7da33342262ccaea4469b4d02b8abb32b6d7bf737f9e0883fece1b8f580

- name: Download tensorrt_yolox/yolox-sPlus-T4-960x960-pseudo-finetune.onnx
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/object_detection_yolox_s/v1/yolox-sPlus-T4-960x960-pseudo-finetune.onnx
dest: "{{ data_dir }}/tensorrt_yolox/yolox-sPlus-T4-960x960-pseudo-finetune.onnx"
mode: "644"
checksum: sha256:f5054e8a890c3be86dc1b4b89a5a36fb2279d4f6110b0159e793be062641bf65

- name: Download tensorrt_yolox/yolox-sPlus-T4-960x960-pseudo-finetune.EntropyV2-calibration.table
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/object_detection_yolox_s/v1/yolox-sPlus-T4-960x960-pseudo-finetune.EntropyV2-calibration.table
dest: "{{ data_dir }}/tensorrt_yolox/yolox-sPlus-T4-960x960-pseudo-finetune.EntropyV2-calibration.table"
mode: "644"
checksum: sha256:cc378d327db5616b0b3a4d077bf37100c25a50ecd22d2b542f54098da100f34c

- name: Download tensorrt_yolox/label.txt
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/label.txt
dest: "{{ data_dir }}/tensorrt_yolox/label.txt"
mode: "644"
checksum: sha256:3540a365bfd6d8afb1b5d8df4ec47f82cb984760d3270c9b41dbbb3422d09a0c

- name: Download tensorrt_yolox/yolox-sPlus-opt-pseudoV2-T4-960x960-T4-seg16cls.onnx
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/object_detection_semseg_yolox_s/v1/yolox-sPlus-opt-pseudoV2-T4-960x960-T4-seg16cls.onnx
dest: "{{ data_dir }}/tensorrt_yolox/yolox-sPlus-opt-pseudoV2-T4-960x960-T4-seg16cls.onnx"
mode: "644"
checksum: sha256:73b3812432cedf65cebf02ca4cb630542fc3b1671c4c0fbf7cee50fa38e416a8

- name: Download tensorrt_yolox/yolox-sPlus-opt-pseudoV2-T4-960x960-T4-seg16cls.EntropyV2-calibration.table
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/object_detection_semseg_yolox_s/v1/yolox-sPlus-opt-pseudoV2-T4-960x960-T4-seg16cls.EntropyV2-calibration.table
dest: "{{ data_dir }}/tensorrt_yolox/yolox-sPlus-opt-pseudoV2-T4-960x960-T4-seg16cls.EntropyV2-calibration.table"
mode: "644"
checksum: sha256:28cd6524d4bcdb2809592a225d28330433e58dc02c92169ea555b44c1a51a584

- name: Download tensorrt_yolox/semseg_color_map.csv
become: true
ansible.builtin.get_url:
url: https://awf.ml.dev.web.auto/perception/models/object_detection_semseg_yolox_s/v1/semseg_color_map.csv
dest: "{{ data_dir }}/tensorrt_yolox/semseg_color_map.csv"
mode: "644"
checksum: sha256:3d93ca05f31b63424d7d7246a01a2365953705a0ed3323ba5b6fddd744a4bfea
41 changes: 7 additions & 34 deletions ansible/roles/ros2/tasks/main.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -28,21 +28,18 @@
ansible.builtin.apt:
name:
- nvidia-jetpack
- nvidia-opencv
- build-essential
- cmake
- git
- libbullet-dev
- libpython3-dev
- python3-wheel
- python3-colcon-common-extensions
- python3-flake8
- python3-pip
- python3-numpy
- python3-pytest-cov
- python3-rosdep
- python3-pytest-covdep
- python3-setuptools
- python3-vcstool
- python3-rosinstall-generator
- libasio-dev
- libtinyxml2-dev
- libcunit1-dev
Expand Down Expand Up @@ -74,6 +71,10 @@
- pytest
- pybind11-global
- importlib_resources
- rosdep
- vcstool
- colcon-common-extensions
- rosinstall-generator
extra_args: -U
become: true # need to sudo privilege to install packages globally

Expand Down Expand Up @@ -209,27 +210,6 @@
path: "{{ source_build_dir }}/jetson-containers"
register: opencv_build_dir_exist

- name: Build and install OpenCV
block:
- name: "{{ block_name }}: Download install script"
ansible.builtin.git:
repo: https://github.com/dusty-nv/jetson-containers.git
dest: "{{ source_build_dir }}/jetson-containers"
version: L4T-R35.1.0
- name: "{{ block_name }}: install dependencies"
shell: ./opencv_install_deps.sh
args:
chdir: "{{ source_build_dir }}/jetson-containers/scripts"
become: true
- name: "{{ block_name }}: execute install script"
shell: ./opencv_install.sh https://nvidia.box.com/shared/static/5v89u6g5rb62fpz4lh0rz531ajo2t5ef.gz OpenCV-4.5.0-aarch64.tar.gz
args:
chdir: "{{ source_build_dir }}/jetson-containers/scripts"
become: true
vars:
block_name: OpenCV
when: not opencv_build_dir_exist.stat.exists

- name: Check YAML build directory existence
ansible.builtin.stat:
path: "{{ source_build_dir }}/yaml-cpp"
Expand Down Expand Up @@ -263,9 +243,9 @@

- name: Upgrade CMake
block:
# Follow directions here: https://apt.kitware.com/
- name: "{{ block_name }}: update gpg key"
shell: wget -O - https://apt.kitware.com/keys/kitware-archive-latest.asc 2>/dev/null | gpg --dearmor - | tee /usr/share/keyrings/kitware-archive-keyring.gpg >/dev/null
# Follow drirections here: https://apt.kitware.com/
- name: "{{ block_name }}: update repository info"
ansible.builtin.apt_repository:
repo: deb [signed-by=/usr/share/keyrings/kitware-archive-keyring.gpg] https://apt.kitware.com/ubuntu/ {{ ansible_distribution_release }} main
Expand All @@ -274,13 +254,6 @@
ansible.builtin.apt:
update_cache: true
state: present
# - name: "{{ block_name }}: delete gpg key"
# ansible.builtin.file:
# path: /usr/share/keyrings/kitware-archive-keyring.gpg
# state: absent
- name: "{{ block_name }}: install kitware-arcyhive-keyring"
ansible.builtin.apt:
name: kitware-archive-keyring
- name: "{{ block_name }}: install newer cmake"
ansible.builtin.apt:
name: cmake
Expand Down
2 changes: 1 addition & 1 deletion ansible/roles/tier4_hdr_camera_driver/defaults/main.yaml
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
camera_driver_repo: https://api.github.com/repos/tier4/tier4_automotive_hdr_camera/releases/latest
camera_driver_repo: https://api.github.com/repos/tier4/tier4_automotive_hdr_camera/releases/142093397
camera_driver_download_dir: /tmp
exposure_time: 11000
13 changes: 7 additions & 6 deletions ansible/setup.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,13 @@
[Warning] Do you want to configure the network? This configuration may overwrite the IP address of the specific network interface [y/N]
private: false
roles:
- role: artifacts
- role: ptp4l_client
when: prompt_configure_network == 'y'
- role: enlarge_txqueue
when: prompt_configure_network == 'y'
- role: netplan
when: prompt_configure_network == 'y'
- role: autoware
- role: cuda
- role: cyclonedds
Expand All @@ -28,9 +35,3 @@
- role: tier4_hdr_camera_driver
when: prompt_install_camera_driver == 'y'
- role: ros2
- role: ptp4l_client
when: prompt_configure_network == 'y'
- role: enlarge_txqueue
when: prompt_configure_network == 'y'
- role: netplan
when: prompt_configure_network == 'y'
8 changes: 8 additions & 0 deletions autoware.repos
Original file line number Diff line number Diff line change
Expand Up @@ -58,3 +58,11 @@ repositories:
type: git
url: https://github.com/tier4/tensorrt_cmake_module.git
version: main
demos:
type: git
url: https://github.com/ros2/demos.git
version: humble
autoware_internal_msgs:
type: git
url: https://github.com/autowarefoundation/autoware_internal_msgs.git
version: main
2 changes: 1 addition & 1 deletion setup-dev-env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,7 @@ export PATH="$HOME/.local/bin:$PATH"
echo -e "\e[36mRunning ansible playbook to setup ECU.\e[0m"

# Run ansible
if ansible-playbook "${SCRIPT_DIR}"/ansible/setup.yaml --ask-become-pass; then
if ansible-playbook "${SCRIPT_DIR}"/ansible/setup.yaml --ask-become-pass --extra-vars data_dir="$HOME"/autoware_data; then
echo -e "\e[32mCompleted.\e[0m"
exit 0
else
Expand Down

0 comments on commit 53c50af

Please sign in to comment.