Commit bfab6bf1 authored by Amir MOHAMMADI's avatar Amir MOHAMMADI

fix frame loading

parent 45a5e8ab
Pipeline #20056 passed with stages
in 81 minutes and 29 seconds
......@@ -18,7 +18,8 @@ class File(VideoPadFile):
:any:`numpy.array`
A frame of the video. The size is (3, 1920, 1080).
"""
vfilename = self.make_path(directory=self.original_directory)
vfilename = self.make_path(
directory=self.original_directory, extension='.avi')
return iter(reader(vfilename))
@property
......@@ -30,7 +31,8 @@ class File(VideoPadFile):
int
The number of frames.
"""
vfilename = self.make_path(directory=self.original_directory)
vfilename = self.make_path(
directory=self.original_directory, extension='.avi')
return reader(vfilename).number_of_frames
@property
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment