Downloads · 30 days
0
bacterial-morphology-resnet34
aaristov/bacterial-morphology-resnet34
bacterial-morphology-resnet34 is a image classification model from aaristov. Use it when you need a label for an image. It is set up for pytorch. The card lists the license as mit.
Multi-label classifier of bacterial droplet morphology from the paper Extending digital biology: bacterial survival and morphological heterogeneity under antibiotic stress (Maikranz et al.). Source data & weights: htt…
Access
Public
Updated Aug 2, 2026
Repo size
85.3 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
Other85.3 MB · 100%
At a glance
- Task
- Image Classification
- Library
- pytorch
- License
- mit
- Access
- Public
- Created
- Aug 2, 2026
- Updated
- Aug 2, 2026
- SHA
- 4f2e1b14
Model card
From the Hugging Face model README
Bacterial Morphology ResNet34
Multi-label classifier of bacterial droplet morphology from the paper Extending digital biology: bacterial survival and morphological heterogeneity under antibiotic stress (Maikranz et al.). Source data & weights: https://doi.org/10.5281/zenodo.17193389 · Code: https://github.com/BaroudLab/Bacterial-Morphology
- Backbone: torchvision
resnet34, final layernn.Linear(512, 5) - Weights file:
bacteria_trained_model_resnet34(plainstate_dict) - Classes (order):
positive, planktonic, clumped, rods, filaments - Multi-label: trained with BCEWithLogitsLoss -> apply sigmoid, threshold 0.5
- Preprocessing:
Grayscale(3) -> Resize(224) -> ToTensor -> Normalize(mean=[0.485,0.456,0.406], std=[0.229,0.224,0.225])
Load
import torch, torchvision
m = torchvision.models.resnet34()
m.fc = torch.nn.Linear(512, 5)
m.load_state_dict(torch.load("bacteria_trained_model_resnet34", map_location="cpu"))
m.eval()
- Task
- Image Classification
- Library
- pytorch
- License
- mit
- Created
- Aug 2, 2026
- Updated
- Aug 2, 2026