Update aflw.py

HRNet · Jun 21, 2019 · 937cf23 · 937cf23
1 parent 8274990
commit 937cf23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/lib/datasets/aflw.py b/lib/datasets/aflw.py
@@ -72,7 +72,7 @@ def __getitem__(self, idx):
                 if random.random() <= 0.6 else 0
             if random.random() <= 0.5 and self.flip:
                 img = np.fliplr(img)
-                pts = fliplr_joints(pts, width=img.shape[1], dataset='aflw')
+                pts = fliplr_joints(pts, width=img.shape[1], dataset='AFLW')
                 center[0] = img.shape[1] - center[0]
 
         img = crop(img, center, scale, self.input_size, rot=r)