Skip to content

Commit

Permalink
update torchvision to 0.18.0+rocm6.0 and facenet-pytorch to 2.5.3
Browse files Browse the repository at this point in the history
  • Loading branch information
rsujeevan committed May 22, 2024
1 parent f7e9f3d commit 9b3d882
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion data/video_to_image.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,7 @@ class VideoToFrames:
Additionally detects all faces for each frame.
"""

_modules_needed_by_name = ["torchvision==0.4.1", "facenet-pytorch==2.2.9"]
_modules_needed_by_name = ["torchvision==0.18.0+rocm6.0", "facenet-pytorch==2.5.3"]

def __init__(self, num_frames_per_video=3, face_additional_area=0.5):
self.num_frames_per_video = num_frames_per_video
Expand Down

0 comments on commit 9b3d882

Please sign in to comment.