Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Workspace with Deeplab #16

Open
wants to merge 8 commits into
base: main
Choose a base branch
from
Open

Conversation

YuZhong-Chen
Copy link
Collaborator

This is the DeepLab pipeline.

Instead of copying the template_ws and installing PyTorch inside it, I made several changes:

- Dockerfile:
    - Changed the base image.
        - I believe installing PyTorch is more challenging than ROS 2 Humble.
    - Manually installed ROS 2 Humble.
- docker-compose:
    - Enabled GPU access.
    - Mounted shared memory.
        - For communication between containers.
    - Mounted PyTorch cache.
        - Keep the models in the cache folder.
Add the CMakeLists.txt and package.xml files to the folder to indicate that it is a ROS package.
The original repository contains lots of training code and utility code.
For our use here, we only need the code for inference.
@YuZhong-Chen YuZhong-Chen requested a review from j3soon October 21, 2023 14:16
@YuZhong-Chen YuZhong-Chen self-assigned this Dec 23, 2023
@j3soon j3soon force-pushed the main branch 2 times, most recently from a8b1110 to dbd12e4 Compare October 23, 2024 19:17
@j3soon j3soon force-pushed the main branch 7 times, most recently from ecf7008 to adf7444 Compare December 15, 2024 03:53
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant