Darknet/YOLO v6.0-79-g2f994075
Object Detection Framework
 
Loading...
Searching...
No Matches
onnx_export_and_verify.py File Reference

Namespaces

namespace  onnx_export_and_verify
 

Functions

 onnx_export_and_verify.darknet_detect (network, image, class_names, threshold)
 
 onnx_export_and_verify.detections_eq (dets_a, dets_b, atol_conf=0.15, atol_box=0.5, threshold=0.75)
 
 onnx_export_and_verify.export_onnx (cfg_file, weights_file, fp16=False)
 
 onnx_export_and_verify.greedy_nearest_match (dets_a, dets_b, atol_conf, atol_box)
 
 onnx_export_and_verify.load_image (image_path, width, height)
 
 onnx_export_and_verify.onnx_detect (session, image, class_names, threshold)
 
 onnx_export_and_verify.single_detection_eq (a, b, atol_conf=0.15, atol_box=0.5)
 

Variables

 onnx_export_and_verify.action
 
bool onnx_export_and_verify.all_eq = True
 
 onnx_export_and_verify.args = parser.parse_args()
 
 onnx_export_and_verify.atol_box
 
 onnx_export_and_verify.atol_conf
 
 onnx_export_and_verify.class_colors = darknet.class_colors(class_names)
 
 onnx_export_and_verify.class_names = open(args.names).read().splitlines()
 
 onnx_export_and_verify.default
 
 onnx_export_and_verify.dets_eq = detections_eq(dn_dets, onnx_dets, atol_conf, atol_box, args.threshold)
 
 onnx_export_and_verify.dn_dets = darknet_detect(dn_network, image, class_names, args.threshold)
 
 onnx_export_and_verify.dn_img = darknet.draw_boxes(dn_dets, image.copy(), class_colors)
 
 onnx_export_and_verify.dn_network
 
 onnx_export_and_verify.float
 
 onnx_export_and_verify.height = darknet.network_height(dn_network)
 
 onnx_export_and_verify.help
 
 onnx_export_and_verify.image = load_image(image_path, width, height)
 
 onnx_export_and_verify.nargs
 
 onnx_export_and_verify.onnx_dets = onnx_detect(onnx_network, image, class_names, args.threshold)
 
 onnx_export_and_verify.onnx_img = darknet.draw_boxes(onnx_dets, image.copy(), class_colors)
 
 onnx_export_and_verify.onnx_model = export_onnx(args.cfg, args.weights, fp16=args.fp16)
 
 onnx_export_and_verify.onnx_network
 
 onnx_export_and_verify.parser = argparse.ArgumentParser(description=__doc__)
 
 onnx_export_and_verify.type
 
 onnx_export_and_verify.width = darknet.network_width(dn_network)