1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
import argparse
import os.path as osp
import matplotlib.pyplot as plt
import imgviz
here = osp.dirname(osp.abspath(__file__))
def run_example(function):
parser = argparse.ArgumentParser(
formatter_class=argparse.ArgumentDefaultsHelpFormatter,
)
parser.add_argument("--save", action="store_true", help="save image")
args = parser.parse_args()
img = function()
if args.save:
out_file = osp.join(here, ".readme/{}.jpg".format(function.__name__))
imgviz.io.imsave(out_file, img)
plt.imshow(img)
plt.axis("off")
plt.show()
|
