diff --git a/tensorlayer/files.py b/tensorlayer/files.py index 96320de01..2300618da 100644 --- a/tensorlayer/files.py +++ b/tensorlayer/files.py @@ -1399,7 +1399,9 @@ def save_joints(): head_x2s = anno['annorect']['x2'][0] head_y2s = anno['annorect']['y2'][0] for annopoint, head_x1, head_y1, head_x2, head_y2 in zip(annopoints, head_x1s, head_y1s, head_x2s, head_y2s): - if annopoint != []: + # if annopoint != []: + # if len(annopoint) != 0: + if annopoint.size: head_rect = [float(head_x1[0, 0]), float(head_y1[0, 0]), float(head_x2[0, 0]), float(head_y2[0, 0])] # joint coordinates