python - StringIO & BytesIO

[junxnone]

Reference

• StringIO和BytesIO

Brief

• StringIO - 在内存中读写 str
• BytesIO - 在内存中读写 Byte

---

• getvalue() 方法用于获得写入后的值

Example

• detectron2 pytorch model --> onnx model

   with torch.no_grad():
        with io.BytesIO() as f:
            torch.onnx.export(
                model,
                inputs,
                f,
                operator_export_type=OperatorExportTypes.ONNX_ATEN_FALLBACK,
                # verbose=True,  # NOTE: uncomment this for debugging
                # export_params=True,
            )
            onnx_model = onnx.load_from_string(f.getvalue())