bharat-b7 / LoopReg

91 stars 10 forks source link

Issue: Running LoopReg on Windows #11

Open codesavory opened 3 years ago

codesavory commented 3 years ago

Hi,

I am trying to run LoopReg on Windows and when running this command -python train_PartSpecificNet.py 1 -mode val -save_name corr -batch_size 16 -split_file assets\data_split_01.pklI get couple of errors - no dataset nuscenes from kaolin and error reading smpl_vt_ft.pklfile. Can you kindly help me with this -


(LoopReg) D:\CG_Source\NeRFs\3D_Avatar_Pipeline\LoopReg>python train_PartSpecificNet.py 1 -mode val -save_name corr -batch_size 16 -split_file assets\data_split_01.pkl
Warning: unable to import datasets/nusc:
   No module named 'nuscenes'
Traceback (most recent call last):
  File "C:\miniconda3\envs\LoopReg\lib\site-packages\kaolin-0.1.0-py3.7-win-amd64.egg\kaolin\datasets\__init__.py", line 11, in <module>
    from .nusc import NuscDetection
  File "C:\miniconda3\envs\LoopReg\lib\site-packages\kaolin-0.1.0-py3.7-win-amd64.egg\kaolin\datasets\nusc.py", line 21, in <module>
    from nuscenes.utils.geometry_utils import transform_matrix
ModuleNotFoundError: No module named 'nuscenes'
Warning: unable to import datasets/nusc:
   None
Using BLB SMPL from the project: LoopReg
Not initializing with pre-trained supervised correspondence network
Traceback (most recent call last):
  File "train_PartSpecificNet.py", line 123, in <module>
    pretrained_path=args.pretrained_path, checkpoint_number=args.checkpoint_number, split_file=args.split_file)
  File "train_PartSpecificNet.py", line 68, in main
    num_workers=16, naked=naked).get_loader(shuffle=False)
  File "D:\CG_Source\NeRFs\3D_Avatar_Pipeline\LoopReg\data_loader\data_loader.py", line 155, in __init__
    self.vt, self.ft = sp.get_vt_ft()
  File "D:\CG_Source\NeRFs\3D_Avatar_Pipeline\LoopReg\lib\smpl_paths.py", line 97, in get_vt_ft
    vt, ft = pkl.load(open(smpl_vt_ft_path, 'r'), encoding='latin-1')
TypeError: a bytes-like object is required, not 'str'