diff --git a/detect.py b/detect.py index 3e17e254..56194118 100644 --- a/detect.py +++ b/detect.py @@ -54,14 +54,6 @@ def detect(save_txt=False, save_img=True, stream_img=False): else: dataloader = LoadImages(opt.source, img_size=img_size, half=opt.half) - # Attempt stream_img: - if stream_img: - try: - cv2.imshow('', np.zeros((480, 640, 3))) - cv2.destroyAllWindows() - except: - stream_img = False # Possible SSH connection, do not stream - # Get classes and colors classes = load_classes(parse_data_cfg(opt.data)['names']) colors = [[random.randint(0, 255) for _ in range(3)] for _ in range(len(classes))] diff --git a/utils/gcp.sh b/utils/gcp.sh index c1f157a2..f56d17ed 100755 --- a/utils/gcp.sh +++ b/utils/gcp.sh @@ -3,7 +3,7 @@ # New VM rm -rf sample_data yolov3 darknet apex coco cocoapi knife knifec git clone https://github.com/ultralytics/yolov3 -# git clone https://github.com/AlexeyAB/darknet && cd darknet && make GPU=1 CUDNN=1 CUDNN_HALF=1 OPENCV=1 && wget -c https://pjreddie.com/media/files/darknet53.conv.74 && cd .. +# git clone https://github.com/AlexeyAB/darknet && cd darknet && make GPU=1 CUDNN=1 CUDNN_HALF=1 OPENCV=0 && wget -c https://pjreddie.com/media/files/darknet53.conv.74 && cd .. git clone https://github.com/NVIDIA/apex && cd apex && pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" . --user && cd .. && rm -rf apex # git clone https://github.com/cocodataset/cocoapi && cd cocoapi/PythonAPI && make && cd ../.. && cp -r cocoapi/PythonAPI/pycocotools yolov3 sudo conda install -y -c conda-forge scikit-image tensorboard pycocotools