samples/dnn: better errormsg in openpose.py

diff --git a/samples/dnn/openpose.py b/samples/dnn/openpose.py
index b79ccd54b80002983a08e5276d13dc6c2bfe77eb..191d23edd4a1b99d826d23dac7279905c71b34b9 100644 (file)
--- a/samples/dnn/openpose.py
+++ b/samples/dnn/openpose.py
@@ -41,8 +41,7 @@ elif args.dataset == 'MPI':
                    ["RElbow", "RWrist"], ["Neck", "LShoulder"], ["LShoulder", "LElbow"],
                    ["LElbow", "LWrist"], ["Neck", "Chest"], ["Chest", "RHip"], ["RHip", "RKnee"],
                    ["RKnee", "RAnkle"], ["Chest", "LHip"], ["LHip", "LKnee"], ["LKnee", "LAnkle"] ]
-else:
-    assert(args.dataset == 'HAND')
+elif args.dataset == 'HAND':
     BODY_PARTS = { "Wrist": 0,
                    "ThumbMetacarpal": 1, "ThumbProximal": 2, "ThumbMiddle": 3, "ThumbDistal": 4,
                    "IndexFingerMetacarpal": 5, "IndexFingerProximal": 6, "IndexFingerMiddle": 7, "IndexFingerDistal": 8,
@@ -61,7 +60,8 @@ else:
                    ["RingFingerProximal", "RingFingerMiddle"], ["RingFingerMiddle", "RingFingerDistal"],
                    ["Wrist", "LittleFingerMetacarpal"], ["LittleFingerMetacarpal", "LittleFingerProximal"],
                    ["LittleFingerProximal", "LittleFingerMiddle"], ["LittleFingerMiddle", "LittleFingerDistal"] ]
-
+else:
+    raise(Exception("you need to specify either 'COCO', 'MPI', or 'Hand' in args.dataset"))
 
 inWidth = args.width
 inHeight = args.height