assert self.embedding_space_ftr.num_embeddings >= idx_occu.shape[0]

[CongSuxu]

assert self.embedding_space_ftr.num_embeddings >= idx_occu.shape[0]
AssertionError

I encounter this error when test on toy_desk dataset. Does anyone know what is causing it?
the error location is in embeding_helper.py line 197.

by the way, how can i change the transformation matrix form slam/opncv format to NDC format, is there any code exist in project that i didnt found.

[ybbbbt]

Hi, CongSuxu,

Can you plese check if #7 can resolve your problem?
If this can't be fixed, please let me know.

BTW, the convension of transformation matrix between SLAM/OpenCV format and NDC format can be found here:

object_nerf/datasets/generic_dataset.py

Lines 358 to 361 in e8b1a7a

	# Original poses has rotation in form "right down forward", change to NDC "right up back"
	fix_rot = np.array([1, 0, 0, 0, -1, 0, 0, 0, -1]).reshape(3, 3)
	pose = np.array(frame["transform_matrix"])
	pose[:3, :3] = pose[:3, :3] @ fix_rot

[CongSuxu]

Thank you for your reply.
I tried #7 's answer that change the N_max_voxels from 800000 to 1200000 (just to enlarge it) in object_nerf_edit_demo_models/toydesk_2/run_config_snapshot.yaml, which is in the downloaded pre-trained models. Then i get the load state_dict error, because the number in checkpoint is 80000, but this number in model is 1200000.

my comand is python test/demo_editable_render.py config=test/config/edit_toy_desk_2.yaml ckpt_path=object_nerf_edit_demo_models/toydesk_2/last.ckpt prefix=toy_desk2_rotating

[ybbbbt]

Hi, it seems that I cannot reproduce this issue. Can you provide more information or logs printed on the terminal? I guess there might be something wrong with checkpoint loading.