← all repositories

binli123/dsmil-wsi

A dual-stream multiple instance learning network using vision transformers and self-supervised learning for classifying whole slide histopathology images and detecting tumors.

465 stars Python Computer VisionDomain Apps
dsmil-wsi
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository provides a PyTorch implementation of DSMIL, a dual-stream multiple instance learning network designed for whole slide image (WSI) classification and tumor detection. The model employs vision transformers and self-supervised contrastive learning to handle weakly supervised learning where only bag-level (slide-level) labels are available. It processes histopathology images by extracting features from image patches and aggregating them using attention-based pooling, enabling tumor classification without pixel-level annotations.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.