Saving figure with bounding boxes

dsgiitr / d2l-pytorch

This project reproduces the book Dive Into Deep Learning (https://d2l.ai/), adapting the code from MXNet into PyTorch.

Apache License 2.0

4.22k stars 1.24k forks source link

Open Anytram opened 2 years ago

Anytram commented 2 years ago

Hi, I was following your tutorial on https://d2l.ai/chapter_computer-vision/bounding-box.html How can I save figure with bounding boxes on it?

suvaansh commented 2 years ago

You may use the "fig.savefig()" command from matplotlib to save the image.