Skip to content

feat: integrate age classification project using torch, numpy, pandas…#31

Closed
jbardowski wants to merge 13 commits intoUMass-Rescue:mainfrom
jbardowski:integrateRB
Closed

feat: integrate age classification project using torch, numpy, pandas…#31
jbardowski wants to merge 13 commits intoUMass-Rescue:mainfrom
jbardowski:integrateRB

Conversation

@jbardowski
Copy link
Copy Markdown

…, requests, opencv-python, sqlalchemy, seaborn, and onnxruntime, with models: cledoux42/Age_Classify_v001 and dima806/fairface_age_image_detection models

…, requests, opencv-python, sqlalchemy, seaborn, and onnxruntime, with models: cledoux42/Age_Classify_v001 and dima806/fairface_age_image_detection models
@prasannals
Copy link
Copy Markdown
Collaborator

Thanks for the PR. Can you please let me know the license for the cledoux42/Age_Classify_v001 and dima806/fairface_age_image_detection models?


class Inputs(TypedDict):
"""Specify the input and output types for the task"""
input_dataset: DirectoryInput
Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

refer https://github.com/UMass-Rescue/RescueBox/pull/28/files,
by overriding the Input types , validation can be added to get a better user experience
comments ?

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I recommend this too. Just fixing the link to !28. #28

PR 28 is now merged and the code example can be found at src/audio-transcription/audio_transcription/main.py

Basically, you'd have to do something like

IMAGE_EXTENSIONS = {".png", ".jpg", ".jpeg", ...}  # mention all allowed extensions here


class ImageDirectory(FileFilterDirectory):
    path: DirectoryPath
    file_extensions: List[str] = IMAGE_EXTENSIONS
    
class Inputs(TypedDict):
    """Specify the input and output types for the task"""

    input_dataset: ImageDirectory  # using ImageDirectory instead of DirectoryInput. This ensures that the directory contains at least one file with at least one of the above listed extensions

@nb950
Copy link
Copy Markdown
Collaborator

nb950 commented Jul 21, 2025

@nb950 nb950 closed this Jul 21, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants