File size: 354 Bytes
c508d7f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import matplotlib.pyplot as plt
plt.switch_backend('agg')
fig = plt.figure()
c1 = plt.Circle((0.2, 0.5), 0.2, color='r')
c2 = plt.Circle((0.8, 0.5), 0.2, color='r')
ax = plt.gca()
ax.add_patch(c1)
ax.add_patch(c2)
plt.axis('scaled')
from tensorboardX import SummaryWriter
writer = SummaryWriter()
writer.add_figure('matplotlib', fig)
writer.close()
|