161b24af716a1e7a3221fce2903544ac1429387d,evaluation/eval_s3dis.py,,,#,10

Before Change

PRED_DIR = os.path.join(BASE_DIR,"upsampling/upsample_pred_A6/seg/")

pred_Areas = os.listdir(PRED_DIR)
for pred_area in pred_Areas:
    pred_Rooms = os.listdir(os.path.join(PRED_DIR,pred_area,"seg"))
    for pred_room in pred_Rooms:
        path_pred_label = os.path.join(PRED_DIR,pred_area,"seg",pred_room)
        pred_label_filenames.append(path_pred_label)

assert(num_room == len(pred_label_filenames))

gt_classes = [0 for _ in range(13)]
positive_classes = [0 for _ in range(13)]

After Change

//pred_data_label_filenames = gt_label_filenames
print(num_room)
print(len(pred_label_filenames))
assert(num_room == len(pred_label_filenames))

gt_classes = [0 for _ in range(13)]

Instances