(args)
| 522 | |
| 523 | |
| 524 | def main(args): |
| 525 | det2_gs = DET2GraphSurgeon(args.exported_onnx, args.det2_config, args.det2_weights) |
| 526 | det2_gs.update_preprocessor(args.batch_size) |
| 527 | anchors = det2_gs.get_anchors(args.sample_image) |
| 528 | det2_gs.process_graph(anchors, args.first_nms_threshold, args.second_nms_threshold) |
| 529 | det2_gs.save(args.onnx) |
| 530 | |
| 531 | |
| 532 | if __name__ == "__main__": |
no test coverage detected