RobbieHolland / SpecialistVLMs

Developing VLMs for expert-level performance in specific medical specialties
3 stars 0 forks source link

some modules are missing #1

Open alisaaalehi opened 1 month ago

alisaaalehi commented 1 month ago

Are you missing the dataset module in your codebase? These are imports in run/closed_ended_evaluation.py script:

import pytorch_lightning as pl
from dataset.retinal_text_dataset import RetinalTextDataset
import copy
...
RobbieHolland commented 1 month ago

Thank you for identifying this. I have added these missing files and updated their import references in the recent commit to main. Let me know if that works for you!

alisaaalehi commented 1 month ago

Thanks, Robbie! I have the missing files now.

It would be also helpful if you could add an "Installation" or "Setup" section to the README (a requirements.yml file, compatible OS, environment variable settings if needed, where the code expects the trained models to be, etc. )

alisaaalehi commented 1 month ago

It seems these modules are missing too: unior_referral_predictions, junior_staging_predictions You've used these in run/closed_ended_figures.py

from evaluation.junior_specialist import junior_referral_predictions, junior_staging_predictions
...