From 9b3d88229cbbe23d08b8bd7d97211209eb933c93 Mon Sep 17 00:00:00 2001 From: Sujeevan Rasaratnam Date: Wed, 22 May 2024 00:26:35 +0000 Subject: [PATCH] update torchvision to 0.18.0+rocm6.0 and facenet-pytorch to 2.5.3 --- data/video_to_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/video_to_image.py b/data/video_to_image.py index 11d03d49..2fdf2f8c 100644 --- a/data/video_to_image.py +++ b/data/video_to_image.py @@ -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