bmartacho / UniPose

We propose UniPose, a unified framework for human pose estimation, based on our “Waterfall” Atrous Spatial Pooling architecture, that achieves state-of-art-results on several pose estimation metrics. Current pose estimation methods utilizing standard CNN architectures heavily rely on statistical postprocessing or predefined anchor poses for joint localization. UniPose incorporates contextual seg- mentation and joint localization to estimate the human pose in a single stage, with high accuracy, without relying on statistical postprocessing methods. The Waterfall module in UniPose leverages the efficiency of progressive filter- ing in the cascade architecture, while maintaining multi- scale fields-of-view comparable to spatial pyramid config- urations. Additionally, our method is extended to UniPose- LSTM for multi-frame processing and achieves state-of-the- art results for temporal pose estimation in Video. Our re- sults on multiple datasets demonstrate that UniPose, with a ResNet backbone and Waterfall module, is a robust and efficient architecture for pose estimation obtaining state-of- the-art results in single person pose detection for both sin- gle images and videos.
Other
211 stars 44 forks source link

ValueError: too many values to unpack (expected 2) ,Hello, how can I solve this problem? #35

Open a139122679 opened 2 years ago

a139122679 commented 2 years ago

Traceback (most recent call last): File "unipose.py", line 270, in trainer = Trainer(args) File "unipose.py", line 64, in init self.val_dir, self.sigma, self.stride, self.workers, self.batch_size) ValueError: too many values to unpack (expected 2)

deeprine commented 2 years ago

utils.py

getDataloader() return train_loader, val_loader

Zinia17 commented 2 years ago

Hi, can any one help me about the following error? Traceback (most recent call last): File "T:\unipose\unipose.py", line 276, in trainer.training(epoch) File "T:\unipose\unipose.py", line 107, in training for i, (input, heatmap, centermap, img_path) in enumerate(tbar): ValueError: too many values to unpack (expected 4)

Zinia17 commented 2 years ago

Another error: Traceback (most recent call last): File "T:\unipose\unipose.py", line 279, in trainer.training(epoch) File "T:\unipose\unipose.py", line 112, in training input_var = input.cuda() AttributeError: 'int' object has no attribute 'cuda'