MikeOfZen / Yet-Another-Openpose-Implementation

This project reimplements from scratch the OpenPose paper (Cao et al,2018), Using Tensorflow 2.1 and optional TPU powered training.
Mozilla Public License 2.0
92 stars 26 forks source link

Bug in post processing code #12

Open sathishram12 opened 1 year ago

sathishram12 commented 1 year ago

On evaluating the model, I used key point detection 2017 validation dataset. On particular image (coco2017/images/val2017/000000182923.jpg), post processing enters in to deadlock mechanism. After debugging, I have found the issue where its present in post.py file

def sort_island_hierarchy(island_hierarchy): """converts the hierarchical island dict to a map to the top level island""" compact = numba.typed.Dict.empty(key_type=numba.types.uint16, value_type=numba.types.uint16) for child, parent in island_hierarchy.items(): last_parent = child while parent != last_parent: last_parent = parent parent = island_hierarchy[parent] compact[child] = last_parent return compact

Lets say i got island_hierarchy as like this {1: 6, 2: 7, 3: 10, 4: 4, 5: 1, 6: 2, 7: 3, 8: 5, 9: 8, 10: 3, 11: 3, 12: 4}. Inner while loop enters in to deadlock mechanism