File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -396,7 +396,7 @@ def get_spatial_dims(
396
396
# Check the validity of each input and add its corresponding spatial_dims to spatial_dims_set
397
397
if boxes is not None :
398
398
if len (boxes .shape ) != 2 :
399
- if boxes .numel () == 0 :
399
+ if boxes .shape [ 0 ] == 0 :
400
400
raise ValueError (
401
401
f"Currently we support only boxes with shape [N,4] or [N,6], "
402
402
f"got boxes with shape { boxes .shape } . "
@@ -413,7 +413,7 @@ def get_spatial_dims(
413
413
spatial_dims_set .add (int (boxes .shape [1 ] / 2 ))
414
414
if points is not None :
415
415
if len (points .shape ) != 2 :
416
- if points .numel () == 0 :
416
+ if points .shape [ 0 ] == 0 :
417
417
raise ValueError (
418
418
f"Currently we support only points with shape [N,2] or [N,3], "
419
419
f"got points with shape { points .shape } . "
You can’t perform that action at this time.
0 commit comments