Open jyh11224 opened 9 months ago
DinoV2 모델 Git clone 및 Dockerfile 생성.
FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
RUN apt-get update -y && apt-get -y install sudo git software-properties-common python3.9 python3-pip libgl1-mesa-glx
RUN mkdir src
WORKDIR /src
RUN git clone https://github.com/Media-4-Machine-Laboratory/DinoV2.git
WORKDIR /src/DinoV2
RUN pip3 install -r requirements.txt
RUN pip3 install -e .
requirements.txt
Jetson-inference 모델 실험 환경 생성 중 오류 발생
github에 접근 불가하여 dockerhub에서 jetson-inference 검색 후 TAG 22.06 이미지 받기
Docker 컨테이너 생성
sudo docker run --gpus all -it --shm-size=8g --ulimit memlock=-1 --ulimit stack=67108864 -e NVIDIA_DRIVER_CAPABILITIES=all dustynv/jetson-inference:22.06
pretrained model 다운
bash download-models.sh
실행 명령어
# C++
$ ./depthnet "images/room_*.jpg" images/test/depth_room_%i.jpg
# Python
./depthnet.py "images/room_*.jpg" images/test/depth_room_%i.jpg
추가 옵션
optional --network
flag which changes the depth model being used (recommended default is fcn-mobilenet
).
optional --visualize
flag which can be comma-separated combinations of input
, depth
--visualize=input,depth
which displays the input and depth images side-by-side--visualize=depth
optional --depth-size
value which scales the size of the depth map relative to the input (the default is 1.0
)
optional --filter-mode
flag which selects point
or linear
filtering used for upsampling (the default is linear
)
optional --colormap
flag which sets the color mapping to use during visualization (the default is viridis_inverted
)도커 컨테이너 사용시 it's recommended to save the output images to the images/test mounted directory. These images will then be easily viewable from your host device under jetson-inference/data/images/test (for more info, see [Mounted Data Volumes].
ssh로 서버 원격 접속 및 vscode 연동