diff --git a/main.py b/main.py index 016aa33..7228f30 100644 --- a/main.py +++ b/main.py @@ -17,8 +17,6 @@ from mmpose.registry import VISUALIZERS from mmpose.structures import merge_data_samples from mmpose.utils import adapt_mmdet_pipeline -from PIL import Image -from torchvision import transforms from util.calc_ste_position import CalcStethoscopePosition from util.ears_ai import EarsAI @@ -101,17 +99,6 @@ print(f"All frames saved to {output_dir}") -def preprocess_image(image_path): - transform = transforms.Compose( - [ - transforms.Resize((224, 224)), - transforms.ToTensor(), - transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]), - ] - ) - return transform(Image.open(image_path).convert("RGB")).unsqueeze(0) - - def extract_keypoints_rtmpose(pose_results): if not pose_results: print("No pose results found.")