Update demo_utils.py

demo_utils.py

@@ -148,6 +148,8 @@ def run_demo(dataset_name, prompt_type, sample_idx, prompt_idx, voxel_size, thet
    #cache
    name_list = [args.dataset, "sample" + str(args.sample_idx), args.prompt_type + "-prompt" + str(args.prompt_idx)]
    name = '_'.join(name_list)
+   os.makedirs("cache_results", exist_ok=True)
+   os.makedirs("cache_prompt", exist_ok=True)
    np.save("./cache_results/" + name + '.npy', new_color)
    np.save("./cache_prompt/" + name + '.npy', PROMPT)
    return new_color, PROMPT