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): ...@@ -18,7 +18,8 @@ class File(VideoPadFile):
:any:`numpy.array` :any:`numpy.array`
A frame of the video. The size is (3, 1920, 1080). 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)) return iter(reader(vfilename))
@property @property
...@@ -30,7 +31,8 @@ class File(VideoPadFile): ...@@ -30,7 +31,8 @@ class File(VideoPadFile):
int int
The number of frames. 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 return reader(vfilename).number_of_frames
@property @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