Skip to content

Commit 120e22b

Browse files
Check average_rate availability in video reader (#8548)
Co-authored-by: Nicolas Hug <[email protected]> Co-authored-by: Nicolas Hug <[email protected]>
1 parent ee6c3d0 commit 120e22b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/io/video_reader.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def get_metadata(self) -> Dict[str, Any]:
251251
rate_n = "framerate"
252252
metadata[stream.type] = {rate_n: [], "duration": []}
253253

254-
rate = stream.average_rate if stream.average_rate is not None else stream.sample_rate
254+
rate = getattr(stream, "average_rate", None) or stream.sample_rate
255255

256256
metadata[stream.type]["duration"].append(float(stream.duration * stream.time_base))
257257
metadata[stream.type][rate_n].append(float(rate))

0 commit comments

Comments
 (0)