• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• connecting_the_dots-master
  • config.json
  • co
    • metric.py
    • gtimer.py
    • plt2d.py
    • plt3d.py
    • io3d.py
    • table.py
    • geometry.py
    • cmap.py
    • __init__.py
    • utils.py
    • args.py
    • plt.py
  • img
  • LICENSE
  • torchext
    • modules.py
    • functions.py
    • setup.py
    • ext
      • ext_cuda.cpp
      • common.h
      • ext.h
      • co_types.h
      • common_cuda.h
      • ext_kernel.cu
      • ext_cpu.cpp
    • __init__.py
    • dataset.py
    • worker.py
  • train_val.py
  • hyperdepth
    • train.cpp
    • hyperdepth.pyx
    • hyperparam_search.py
    • hyperdepth.h
    • setup.py
    • eval.cpp
    • vis_eval.py
    • rf
      • common.h
      • serialization.h
      • splitfcn.h
      • data.h
      • tree.h
      • node.h
      • spliteval.h
      • train.h
      • threadpool.h
      • leaffcn.h
      • forest.h
  • data
    • commons.py
    • lcn
      • test_lcn.py
      • lcn.pyx
      • readme.txt
      • setup.py
      • lcn.html
    • create_syn_data.py
    • __init__.py
    • dataset.py
  • readme.md
  • requirements.txt
  • model
    • exp_synph.py
    • exp_synphge.py
    • __init__.py
    • networks.py
  • renderer
    • Makefile
    • cyrender.pyx
    • render
      • common.h
      • render_gpu_dummy.cpp
      • stdlib_cuda.cu
      • render_gpu.cu
      • stdlib_cuda.h
      • stdlib_cuda_dummy.cpp
      • render_gpu.h
      • render_cpu.h
      • co_types.h
      • common_cpu.h
      • render_cpu.cpp
      • render.h
      • geometry.h
      • common_cuda.h
    • setup.py
    • __init__.py
  • create_syn_data.sh

import numpy as np
import matplotlib as mpl
from matplotlib import _pylab_helpers
from matplotlib.rcsetup import interactive_bk as _interactive_bk
import matplotlib.pyplot as plt
import os
import time

def save(path, remove_axis=False, dpi=300, fig=None):
  if fig is None:
    fig = plt.gcf()
  dirname = os.path.dirname(path)
  if dirname != '' and not os.path.exists(dirname):
    os.makedirs(dirname)
  if remove_axis:
    for ax in fig.axes:
      ax.axis('off')
      ax.margins(0,0)
    fig.subplots_adjust(top=1, bottom=0, right=1, left=0, hspace=0, wspace=0)
    for ax in fig.axes:
      ax.xaxis.set_major_locator(plt.NullLocator())
      ax.yaxis.set_major_locator(plt.NullLocator())
  fig.savefig(path, dpi=dpi, bbox_inches='tight', pad_inches=0)

def color_map(im_, cmap='viridis', vmin=None, vmax=None):
  cm = plt.get_cmap(cmap)
  im = im_.copy()
  if vmin is None:
    vmin = np.nanmin(im)
  if vmax is None:
    vmax = np.nanmax(im)
  mask = np.logical_not(np.isfinite(im))
  im[mask] = vmin
  im = (im.clip(vmin, vmax) - vmin) / (vmax - vmin)
  im = cm(im)
  im = im[...,:3]
  for c in range(3):
    im[mask, c] = 1
  return im

def interactive_legend(leg=None, fig=None, all_axes=True):
  if leg is None:
    leg = plt.legend()
  if fig is None:
    fig = plt.gcf()
  if all_axes:
    axs = fig.get_axes()
  else:
    axs = [fig.gca()]

  # lined = dict()
  # lines = ax.lines
  # for legline, origline in zip(leg.get_lines(), ax.lines):
  #   legline.set_picker(5)
  #   lined[legline] = origline
  lined = dict()
  for lidx, legline in enumerate(leg.get_lines()):
    legline.set_picker(5)
    lined[legline] = [ax.lines[lidx] for ax in axs]

  def onpick(event):
    if event.mouseevent.dblclick:
      tmp = [(k,v) for k,v in lined.items()]
    else:
      tmp = [(event.artist, lined[event.artist])]

    for legline, origline in tmp:
      for ol in origline:
        vis = not ol.get_visible()
        ol.set_visible(vis)
      if vis:
        legline.set_alpha(1.0)
      else:
        legline.set_alpha(0.2)
    fig.canvas.draw()

  fig.canvas.mpl_connect('pick_event', onpick)

def non_annoying_pause(interval, focus_figure=False):
  # https://github.com/matplotlib/matplotlib/issues/11131
  backend = mpl.rcParams['backend']
  if backend in _interactive_bk:
    figManager = _pylab_helpers.Gcf.get_active()
    if figManager is not None:
      canvas = figManager.canvas
      if canvas.figure.stale:
        canvas.draw()
      if focus_figure:
        plt.show(block=False)
      canvas.start_event_loop(interval)
      return
  time.sleep(interval)

def remove_all_ticks(fig=None):
  if fig is None:
    fig = plt.gcf()
  for ax in fig.axes:
    ax.axes.get_xaxis().set_visible(False)
    ax.axes.get_yaxis().set_visible(False)