From 7fe5e135aa4f0f9e74845b74bd5dfb536ac35df3 Mon Sep 17 00:00:00 2001
From: Victor Sonck <victor.sonck@gmail.com>
Date: Sun, 31 Jan 2021 03:52:14 +0100
Subject: [PATCH] Fix dockerfile bionic opencv 3.4 version (#383)

* Fix dockerfile bionic opencv 3.4 version

* add libraries needed for yolov5

* add libshape debug package

Co-authored-by: Victor Sonck <victor.sonck@ml6.eu>
---
 Dockerfile | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/Dockerfile b/Dockerfile
index 8f2705d..bb4fe61 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -3,7 +3,13 @@ ARG CUDA="10"
 
 FROM hakuyyf/tensorrtx:trt${TENSORRT}_cuda${CUDA}
 
+# Get opencv 3.4 for bionic based images
+RUN rm /etc/apt/sources.list.d/timsc-ubuntu-opencv-3_3-bionic.list
+RUN rm /etc/apt/sources.list.d/timsc-ubuntu-opencv-3_3-bionic.list.save
+RUN add-apt-repository -y ppa:timsc/opencv-3.4
+
 RUN apt-get update
+RUN apt-get install -y libopencv-dev libopencv-dnn-dev libopencv-shape3.4-dbg
 
 # git clone tensorrtx
 RUN git clone https://github.com/wang-xinyu/tensorrtx.git
\ No newline at end of file