def ocl_setup(): have_ocl = haveOpenCL() info("have_ocl = %s" % have_ocl) if have_ocl: if not useOpenCL(): setUseOpenCL(True) info("useOpenCL = %s" % useOpenCL()) info("haveAmdBlas() = %s" % haveAmdBlas()) info("haveAmdFft() = %s" % haveAmdFft())
import time import datetime import os import numpy as np import image_processing as improc import math_operation as mo import _vehicle_init as vehicleInit import _trajectory_init as trajectoryInit import shadow_removal as sr from PyQt4 import QtGui, QtCore from PyQt4 import uic from cv2 import ocl from munkres import Munkres ocl.setUseOpenCL(False) # set flag OCL to False if you build OPENCV -D WITH_OPENCL=ON class QtCapture: def setVideoMode(self, video_mode): self.video_mode = video_mode def getVideoMode(self): return self.video_mode def setVideoOutput(self, video_output): self.video_output = video_output def getVideoOutput(self): return self.video_output def setBackgroundSubtraction(self, backgroundSubtraction):
for key in t.keys(): if key not in T: T[key] = [] P[key] = [] T[key].extend(t[key]) P[key].extend(p[key]) output_ap(T, P) except AssertionError as e: _, _, tb = sys.exc_info() traceback.print_tb(tb) # Fixed format tb_info = traceback.extract_tb(tb) filename, line, func, text = tb_info[-1] print('An error occurred on line {} in statement {}'.format( line, text)) except Exception as e: print(f"Some exception occur!\n\r {e} \n we don't care") TP_obj = [T, P, T_real, P_real] with open(measure_mAP_filename, "wb") as f_measure_mAP: pickle.dump(TP_obj, f_measure_mAP) draw_measure_curve(T, P, T_real, P_real, original_mapping) sess.close() if __name__ == '__main__': ocl.setUseOpenCL(True) _main(parser.parse_args())