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,