From a8e025f7a86ff5a4161c30718e0400ab982b04e8 Mon Sep 17 00:00:00 2001 From: matteo porta Date: Mon, 18 Jul 2022 15:52:23 +0200 Subject: [PATCH] fix nogpu --- src/components/vision.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/src/components/vision.py b/src/components/vision.py index b489bb0..5bb7560 100644 --- a/src/components/vision.py +++ b/src/components/vision.py @@ -7,8 +7,6 @@ from pathlib import Path import cv2 import numpy as np -import tensorflow as tf -from lib.helpers.object_detection.utils import label_map_util from PyQt5.QtCore import (QFileSystemWatcher, QLineF, QMutex, QPointF, QRectF, Qt, QThread, pyqtSignal) from PyQt5.QtGui import (QBrush, QColor, QFont, QImage, QPainter, QPainterPath, @@ -17,6 +15,12 @@ from PyQt5.QtGui import (QBrush, QColor, QFont, QImage, QPainter, QPainterPath, from .component import Component from .consumer import Consumer +if "--no-gpu" in sys.argv: + os.environ["CUDA_VISIBLE_DEVICES"] = "-1" + +import tensorflow as tf +from lib.helpers.object_detection.utils import label_map_util + if "--no-edgetpu" not in sys.argv: try: from pycoral.utils.edgetpu import make_interpreter @@ -42,9 +46,6 @@ else: def Interpreter(*args, **kwargs): raise ValueError("\"--no-tflite\" in sys.argv") -if "--no-gpu" in sys.argv: - os.environ["CUDA_VISIBLE_DEVICES"] = "-1" - # # Patch the location of gfile # tf.gfile = tf.io.gfile