Atlas
Atlas copied to clipboard
model weights are not loaded
Good day. Model weights are not loaded.
import os
import numpy as np
import torch
from atlas.data import SceneDataset, parse_splits_list
from atlas.model import VoxelNet
import atlas.transforms as transforms
model = VoxelNet
model.load_from_checkpoint(checkpoint_path='final.ckpt')
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-6-89c83cb76361> in <module>()
----> 1 model.load_from_checkpoint(checkpoint_path='final.ckpt')
/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/pytorch_lightning/core/saving.py in load_from_checkpoint(cls, checkpoint_path, map_location, hparams_file, strict, **kwargs)
155 checkpoint[cls.CHECKPOINT_HYPER_PARAMS_KEY].update(kwargs)
156
--> 157 model = cls._load_model_state(checkpoint, strict=strict, **kwargs)
158 return model
159
/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/pytorch_lightning/core/saving.py in _load_model_state(cls, checkpoint, strict, **cls_kwargs_new)
197 _cls_kwargs = {k: v for k, v in _cls_kwargs.items() if k in cls_init_args_name}
198
--> 199 model = cls(**_cls_kwargs)
200
201 # give model a chance to load something
/home/ubuntu/kot/Atlas/atlas/model.py in __init__(self, hparams)
91
92 # see config.py for details
---> 93 self.hparams = hparams
94
95 # pytorch lightning does not support saving YACS CfgNone
/home/ubuntu/anaconda3/envs/pytorch_p36/lib/python3.6/site-packages/torch/nn/modules/module.py in __setattr__(self, name, value)
636 buffers[name] = value
637 else:
--> 638 object.__setattr__(self, name, value)
639
640 def __delattr__(self, name):
AttributeError: can't set attribute
You can tell what could be wrong. I use the environment as required.
In Colab notebook exactly the same
It's either a typo (self.hparams
should be self._hparams
in the initialization function) or the authors forgot to implement the setter function for the self.hparams
property (in which the property itself is defined in the parent class, but the setter function needs to be defined the each sub-class)
Getting the same issue, after spending hours to prep the environment...
@zmurez-ml , any idea on how to sort this out?
tks
So in model.py, line 93, replacing self.hparams = hparams
by self._hparams = hparams
, it fixes the issue.
Thanks @dephaniehe.
Attaching an inference with voxel size of 224 x 224 x 64 (do not have enough RAM for more). Quality is ok, although a stereo camera with a good SLAM algo does a better job I think. Would love to have the textures, need to look at that now.
So in model.py, line 93, replacing
self.hparams = hparams
byself._hparams = hparams
, it fixes the issue. Thanks @dephaniehe.Attaching an inference with voxel size of 224 x 224 x 64 (do not have enough RAM for more). Quality is ok, although a stereo camera with a good SLAM algo does a better job I think. Would love to have the textures, need to look at that now.
Hi @Tetsujinfr , sorry to bother. May I know which visualization app u are using? I tried cloudcompare and meshlab but the inference results as in the readme seems not ok for these two.
From memory I think I have used this online viewer. I just checked and it works with the .ply generated on my side (no texture).
Hi,I can run inference.py and output three files (samples1.npz, sample1.ply, sample1_attributes.npz ) with your provided data. I want to visulize sample1.ply so I use colmap, but it get error. could you give me some suggestions about how to visulize the results from model output?