From 0f2588d86b99f3354df0564c1422bc3e2bd6844f Mon Sep 17 00:00:00 2001 From: Yudai Kato Date: Wed, 24 Aug 2022 06:05:06 +0000 Subject: [PATCH] fix bugs to run training script --- docker/Dockerfile | 5 +++++ tridet/visualizers/box3d_visualizer.py | 4 ++-- 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index e81809e..d5c238f 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -27,6 +27,11 @@ ENV WANDB_API_KEY=${WANDB_API_KEY} # Install core APT packages. # ------------------------- ENV DEBIAN_FRONTEND=noninteractive + +# avoid GPG error +RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/3bf863cc.pub +RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub + RUN apt-get update && apt-get install -y \ # essential build-essential \ diff --git a/tridet/visualizers/box3d_visualizer.py b/tridet/visualizers/box3d_visualizer.py index 2b5f2a5..1f15c32 100644 --- a/tridet/visualizers/box3d_visualizer.py +++ b/tridet/visualizers/box3d_visualizer.py @@ -339,7 +339,7 @@ def visualize(self, x): # By default, ["forward" of body] == ["right" of BEV image]. # Change it to ["up" of BEV image] by rotating. - viz_image = cv2.rotate(viz_image, cv2.cv2.ROTATE_90_COUNTERCLOCKWISE) + viz_image = cv2.rotate(viz_image, cv2.ROTATE_90_COUNTERCLOCKWISE) # Crop the BEV image to show only frustum. viz_image = bev_frustum_crop(viz_image) @@ -382,7 +382,7 @@ def visualize(self, x): ) # By default, ["forward" of body] == ["right" of BEV image]. # Change it to ["up" of BEV image] by rotating. - viz_image = cv2.rotate(viz_image, cv2.cv2.ROTATE_90_COUNTERCLOCKWISE) + viz_image = cv2.rotate(viz_image, cv2.ROTATE_90_COUNTERCLOCKWISE) # Crop the BEV image to show only frustum. viz_image = bev_frustum_crop(viz_image)