cassiePython/CLIPNeRF

module 'kornia' has no attribute 'rgb_to_grayscale'

ZYY-666 opened this issue · 5 comments

It seems that the version of my kornia is not the same as the one you use, so can you tell me your kornia version?

This problom occurs in File "run_nerf_clip.py", line 808, in train:
rgb_img_gray = kornia.rgb_to_grayscale(rgb_img)

try this
kornia.color.rgb_to_grayscale()

imageio~=2.21.1
imageio-ffmpeg
matplotlib
configargparse
tensorboard>=2.0
tqdm~=4.64.0
opencv-python~=4.6.0.66

torch~=1.8.2+cu111
clip~=1.0
numpy~=1.23.2
kornia~=0.6.6

can you update the requirements? so that all packages are included?

@ZYY-666 does the training work for you and result in the images from the repository?
#9 (comment)

for me the loss is not going down... and the resulting images look rather laggy

try this kornia.color.rgb_to_grayscale()

Thanks. this works for mE!