utils.vis_utils 保存图片
时间: 2023-09-22 09:10:52 浏览: 40
utils.vis_utils 保存图片的方法可以用以下代码实现:
```python
import matplotlib.pyplot as plt
def save_fig(fig, file_path):
"""
保存 matplotlib figure 对象到指定路径
:param fig: matplotlib figure 对象
:param file_path: 图片保存路径,包含文件名和格式,例如:'./test.png'
"""
fig.savefig(file_path)
```
其中,`fig` 是 matplotlib 中的 figure 对象,`file_path` 是图片保存路径,可以是相对路径或绝对路径,包含文件名和格式,例如:'./test.png'。
调用该方法时,可以将 matplotlib 中的 figure 对象传入,同时指定图片保存路径,例如:
```python
fig, ax = plt.subplots()
ax.plot([1, 2, 3], [4, 5, 6])
save_fig(fig, './test.png')
```
这段代码会绘制一条直线并将 figure 对象保存为 './test.png'。