vis_benchmark.py

import os
import sys
import os.path as osp
from tf_lib.pc_viz import visualize_pointcloud
import argparse

parser = argparse.ArgumentParser(description='Qualitative comparision of ResGCN '
                                             'against PlainGCN on PartNet segmentation')

parser.add_argument('--pointsize', default=3, type=int, help='point size')
parser.add_argument('--input_path',
                    default=None,
                    type=str,
                    help='path to the input')
parser.add_argument('--gt_path',
                    default=None,
                    type=str,
                    help='path to the gt')
parser.add_argument('--punet_path',
                    default=None,
                    type=str, help='path to the result')
parser.add_argument('--mpu_path',
                    default=None,
                    type=str, help='path to the result')
parser.add_argument('--pugan_path',
                    default=None,
                    type=str, help='path to the result')
parser.add_argument('--pugcn_path',
                    default=None,
                    type=str, help='path to the result')
parser.add_argument('--png_path',
                    default=None, required=True,
                    type=str, help='name of file to show')
parser.add_argument('--filename',
                    default=None, required=True,
                    type=str, help='name of file to show')

args = parser.parse_args()

comparison_folder_list = [args.input_path]
text = ['Input']
if args.punet_path is not None:
    comparison_folder_list.append(args.punet_path)
    text.append('PU-Net')
if args.mpu_path is not None:
    comparison_folder_list.append(args.mpu_path)
    text.append('3PU')
if args.pugan_path is not None:
    comparison_folder_list.append(args.pugan_path)
    text.append('PU-GAN')
if args.pugcn_path is not None:
    comparison_folder_list.append(args.pugcn_path)
    text.append('pugcn_path')
if args.gt_path is not None:
    comparison_folder_list.append(args.gt_path)
    text.append('Ground Truth')

filename = args.filename
png_path = osp.join(args.png_path, filename.replace(filename.split('.')[-1], 'png'))
# color = [22, 139, 119]
color = [0, 0, 0]

visualize_pointcloud(comparison_folder_list,
                     filename,
                     color,
                     text=text,
                     png_path=png_path,
                     interactive=True,
                     orientation='horizontal',
                     pointsize=args.pointsize
                     )