LesNet (Lesion Net) is an open-source project for AI-based skin lesion detection. It aims to create a reliable tool and foster community involvement in critical AI problems. Contributions are welcome!
This PR introduces a series of enhancements, architectural changes, and optimizations to the project. Significant efforts have been made to improve model performance, refine learning rates, and structure application components for better maintainability.
Key Changes
๐ Reduced model complexity for efficiency while improving performance.
โก๏ธ Updated learning rates to mitigate initial training overshoots.
๐ก Integrated Focal Loss function to bolster training on challenging classifications.
๐ฏ Refined metric priorities to emphasize Recall and AUC, aligning evaluation with diagnostic significance.
๐ Introduced a new command scripts setup for streamlined data scraping.
๐ Overhauled application architecture, organizing command-related files into a dedicated commands folder for clarity.
๐ Implemented complex residual model architectures to enhance detection performance.
๐งช Added a Hyperparameter Tuning Module, leveraging TensorBoard for detailed performance tracking.
Structural Enhancements
๐ Restructured the application for simplicity, moving operational commands into a centralized commands folder.
๐ Updated architecture to include a model layer, enhanced service layer, and operational commands, removing outdated 'core' and 'ai' directories.
Optimizations & Bug Fixes
๐ง Improved data augmentation and class balance handling for more effective training.
๐ Fixed minor bugs, including a fix for the model saving format and test case updates.
This PR introduces a series of enhancements, architectural changes, and optimizations to the project. Significant efforts have been made to improve model performance, refine learning rates, and structure application components for better maintainability.
Key Changes
commands
folder for clarity.Structural Enhancements
commands
folder.Optimizations & Bug Fixes