Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cell Finetuned Model + Tests + OOP #3

Merged
merged 11 commits into from
Apr 23, 2024

Conversation

ian-coccimiglio
Copy link
Contributor

I put in a new finetuned object detection model and recipe called "ObjectAwareModel_Cell_FT" and "Mobile SAM Finetuned" respectively. It has a FasterRCNN_MobileNet backbone, and was finetuned on the entirety of the Cellpose training images and labels (after converting the ground-truth segmentations to bounding boxes) for a few epochs until I got tired of waiting. Still probably plenty of room to optimize, but it's a start.

The model performs respectably well on "cellular" images, whereas the original MobileSAMv2 (pretrained YOLO) model would often fail. However, it simultaneously performs worse on "general" images.

I also changed part of the code structure to be more object-oriented by creating some classes for object detectors. This was because I was starting to add repetitive functions every time I wanted to work on a new model. However, the API remains nearly the same, with get_bounding_boxes as being the major driver of object_detection, but now you need to select the detector_model between "YOLOv8" and "Finetuned". Names are only placeholders.

@bnorthan
Copy link
Contributor

Thanks - this project is really starting to look great.

@bnorthan bnorthan merged commit 5b9c334 into True-North-Intelligent-Algorithms:main Apr 23, 2024
1 of 10 checks passed
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.

None yet

2 participants