def __init__(self): config = Config() self.ts_queue = queue.Queue() self.max_threads = Response().get_max_threads() self.video_format = config.getValue("video_format") self.ffmpeg_path = config.getValue("ffmpeg_path") self.key_host = Response().get_key_url()
EmbeddingLoss, primitive_loss, evaluate_miou, ) from src.segment_utils import to_one_hot, SIOU_matched_segments from src.utils import visualize_point_cloud_from_labels, visualize_point_cloud from src.dataset import generator_iter from src.mean_shift import MeanShift from src.segment_utils import SIOU_matched_segments from src.residual_utils import Evaluation import time from src.primitives import SaveParameters # Use only one gpu. os.environ["CUDA_VISIBLE_DEVICES"] = "0" config = Config(sys.argv[1]) if_normals = config.normals userspace = "" Loss = EmbeddingLoss(margin=1.0) if config.mode == 0: # Just using points for training model = PrimitivesEmbeddingDGCNGn( embedding=True, emb_size=128, primitives=True, num_primitives=10, loss_function=Loss.triplet_loss, mode=config.mode, num_channels=3,