diff --git a/yolo3_one_file_to_detect_them_all.py b/yolo3_one_file_to_detect_them_all.py index 17ca64e22..3df17bf74 100644 --- a/yolo3_one_file_to_detect_them_all.py +++ b/yolo3_one_file_to_detect_them_all.py @@ -1,5 +1,5 @@ import argparse -import os +import os, sys import numpy as np from keras.layers import Conv2D, Input, BatchNormalization, LeakyReLU, ZeroPadding2D, UpSampling2D from keras.layers.merge import add, concatenate @@ -7,7 +7,7 @@ import struct import cv2 -np.set_printoptions(threshold=np.nan) +np.set_printoptions(threshold=sys.maxsize) os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"]="0"