Support inference yaml saving for export (#492)

* update downloading

* support inference yml saving for export

Author: warrentdrew

requirements.txt

@@ -6,7 +6,6 @@ numpy
 nuscenes-devkit
 opencv-python <= 4.6.0.66
 opencv-python-headless <= 4.6.0.66
-open3d
 pandas
 paddledet
 paddleseg

requirements_pdx.txt

@@ -6,7 +6,6 @@ numpy
 nuscenes-devkit
 opencv-python <= 4.6.0.66
 opencv-python-headless == 4.10.0.84
-open3d
 pandas
 paddledet
 paddleseg

tools/export.py

@@ -23,6 +23,7 @@
 from paddle3d.slim import get_qat_config
 from paddle3d.utils.checkpoint import load_pretrained_model
 from paddle3d.utils.logger import logger
+from paddle3d.utils.save_result import dump_infer_config
 
 parser = argparse.ArgumentParser(description='Model Export')
 
@@ -66,6 +67,11 @@ def parse_normal_args():
         help='Config for quant model.',
         default=None,
         type=str)
+    parser.add_argument(
+        '--save_inference_yml',
+        dest='save_inference_yml',
+        help='Whether to save inference yml for exported models.',
+        action='store_true')
 
     return parser.parse_known_args()
 
@@ -155,6 +161,10 @@ def main(args, rest_args):
 
     model.export(args.save_dir, name=args.save_name, **kwargs)
 
+    if args.save_inference_yml:
+        dump_infer_config(cfg.pdx_cfg,
+                          os.path.join(args.save_dir, 'inference.yml'))
+
     if args.export_for_apollo:
         if not isinstance(model, BaseDetectionModel):
             logger.error('Model {} does not support Apollo yet!'.format(