(args)
| 418 | |
| 419 | |
| 420 | def main(args): |
| 421 | effdet_gs = EfficientDetGraphSurgeon(args.saved_model) |
| 422 | if args.tf2onnx: |
| 423 | effdet_gs.save(args.tf2onnx) |
| 424 | effdet_gs.update_preprocessor(args.input_format, args.input_size, args.preprocessor) |
| 425 | effdet_gs.update_shapes() |
| 426 | effdet_gs.update_network() |
| 427 | effdet_gs.update_nms(args.nms_threshold, args.nms_detections) |
| 428 | effdet_gs.save(args.onnx) |
| 429 | |
| 430 | |
| 431 | if __name__ == "__main__": |
no test coverage detected