search

DLR-RM / AugmentedAutoencoder

Official Code: Implicit 3D Orientation Learning for 6D Object Detection from RGB Images
MIT License
339 stars 97 forks source link

ValueError: need more than 3 values to un pack #7

Closed jinjuehui closed 5 years ago

jinjuehui commented 5 years ago

Hi, after I create the training config file, and insert the paths of my 3d model, when I run "ae_train exp_group/my_autoencoder -d" I get following ValueError :

128 128 3 [[8, 8], [16, 16], [32, 32], [64, 64]] (?, 128, 128, 3) (?, 128, 128, 3) Traceback (most recent call last):20000 ] | | (ETA: --:--:--) File "/localhome/demo/autoencoder_6d_pose_estimation/venv/bin/ae_train", line 11, in load_entry_point('auto-pose==0.9', 'console_scripts', 'ae_train')() File "/localhome/demo/autoencoder_6d_pose_estimation/venv/local/lib/python2.7/site-packages/auto_pose/ae/ae_train.py", line 90, in main dataset.get_training_images(dataset_path, args) File "/localhome/demo/autoencoder_6d_pose_estimation/venv/local/lib/python2.7/site-packages/auto_pose/ae/dataset.py", line 93, in get_training_images self.render_training_images() File "/localhome/demo/autoencoder_6d_pose_estimation/venv/local/lib/python2.7/site-packages/auto_pose/ae/dataset.py", line 246, in render_training_images bgr_x, depth_x = self.renderer.render( File "/localhome/demo/autoencoder_6d_pose_estimation/venv/local/lib/python2.7/site-packages/auto_pose/ae/utils.py", line 15, in decorator setattr(self, attribute, function(self)) File "/localhome/demo/autoencoder_6d_pose_estimation/venv/local/lib/python2.7/site-packages/auto_pose/ae/dataset.py", line 75, in renderer float(self._kw['vertex_scale']) File "/localhome/demo/autoencoder_6d_pose_estimation/venv/local/lib/python2.7/site-packages/auto_pose/meshrenderer/meshrenderer_phong.py", line 48, in init for vertex, normal, color, faces in attributes: ValueError: need more than 3 values to unpack

It' look like the attributes are not correctly loaded, as I check the attributes in debug mode the size of attributes is 3 not 4. The 3d model I use is T-LESS models_cad/obj_01.ply. I tried other .ply meshes and have also problems. the background image: VOCdevkit/VOC2012/JPEGImages/'.jpg

System Info My environment: I'm using pycharm virtualenvironment

thanx..

MartinSmeyer commented 5 years ago

yes, you have to change the model type in the config file from "reconst" to "cad" if you use CAD models for training:

MODEL: reconst to MODEL: cad

jinjuehui commented 5 years ago

it works, thanks a lot!

eyildiz-ugoe commented 5 years ago

@MartinSmeyer This error occurs whichever value I enter, cad or reconst. Please see https://github.com/DLR-RM/AugmentedAutoencoder/issues/9