The dataset.json leaves the original extension, but the file is converted into a jpeg which results in a keyerror
Loading networks from "/content/PanoHead/models/easy-khair-180-gpc0.8-trans10-025000.pkl"... Traceback (most recent call last): File "/content/PanoHead/projector_withseg.py", line 439, in <module> run_projection() # pylint: disable=no-value-for-parameter File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 1130, in __call__ return self.main(*args, **kwargs) File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 1055, in main rv = self.invoke(ctx) File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 1404, in invoke return ctx.invoke(self.callback, **ctx.params) File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 760, in invoke return __callback(*args, **kwargs) File "/content/PanoHead/projector_withseg.py", line 325, in run_projection c = torch.from_numpy(dataset._get_raw_labels()[idx:idx+1]).to(device) File "/content/PanoHead/training/dataset.py", line 59, in _get_raw_labels self._raw_labels = self._load_raw_labels() if self._use_labels else None File "/content/PanoHead/training/dataset.py", line 244, in _load_raw_labels labels = [labels[fname.replace('\\', '/')] for fname in self._image_fnames] File "/content/PanoHead/training/dataset.py", line 244, in <listcomp> labels = [labels[fname.replace('\\', '/')] for fname in self._image_fnames] KeyError: '[YOURFILENAME].jpg'
The dataset.json leaves the original extension, but the file is converted into a jpeg which results in a keyerror
Loading networks from "/content/PanoHead/models/easy-khair-180-gpc0.8-trans10-025000.pkl"... Traceback (most recent call last): File "/content/PanoHead/projector_withseg.py", line 439, in <module> run_projection() # pylint: disable=no-value-for-parameter File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 1130, in __call__ return self.main(*args, **kwargs) File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 1055, in main rv = self.invoke(ctx) File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 1404, in invoke return ctx.invoke(self.callback, **ctx.params) File "/usr/local/lib/python3.10/dist-packages/click/core.py", line 760, in invoke return __callback(*args, **kwargs) File "/content/PanoHead/projector_withseg.py", line 325, in run_projection c = torch.from_numpy(dataset._get_raw_labels()[idx:idx+1]).to(device) File "/content/PanoHead/training/dataset.py", line 59, in _get_raw_labels self._raw_labels = self._load_raw_labels() if self._use_labels else None File "/content/PanoHead/training/dataset.py", line 244, in _load_raw_labels labels = [labels[fname.replace('\\', '/')] for fname in self._image_fnames] File "/content/PanoHead/training/dataset.py", line 244, in <listcomp> labels = [labels[fname.replace('\\', '/')] for fname in self._image_fnames] KeyError: '[YOURFILENAME].jpg'