Skip to content

Commit ad317a6

Browse files
Harshith ReddyHarshith Reddy
authored andcommitted
Add README images (architecture, results, qualitative)
1 parent b24cf78 commit ad317a6

5 files changed

Lines changed: 14 additions & 0 deletions

File tree

README.md

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,8 @@ Colorectal cancer (CRC) is the second leading cause of cancer-related death worl
1515

1616
Encoder-decoder with ResNet50 encoder, four levels of feature maps, DCP (dilated convolution pooling) blocks, and four decoder blocks with channel and spatial attention. Two model variants: **Kvasir-SEG** and **BKAI-IGH**.
1717

18+
![Architecture](images/architecture.jpg)
19+
1820
### Implementation
1921

2022
- **Framework:** PyTorch (1.9.0+), implemented with a single GeForce RTX 3090 GPU (24 GB).
@@ -26,6 +28,18 @@ Encoder-decoder with ResNet50 encoder, four levels of feature maps, DCP (dilated
2628
- **Metrics:** Dice **0.90**, mIoU **0.83**, ~33.68 FPS on GPU.
2729
- Qualitative results and heatmaps show explanation for polyp location, improving trustworthiness of the method.
2830

31+
**Quantitative results (same dataset):**
32+
33+
![Result 1](images/result-1.png)
34+
35+
**Quantitative results (different dataset):**
36+
37+
![Result 2](images/result-2.png)
38+
39+
**Qualitative results:**
40+
41+
![Qualitative](images/qualitative.jpg)
42+
2943
### Repo structure
3044

3145
- **Frontend:** React app (this repo root) — upload colonoscopy image, choose model (Kvasir-SEG or BKAI-IGH), view segmentation mask and overlay.

images/architecture.jpg

148 KB
Loading

images/qualitative.jpg

239 KB
Loading

images/result-1.png

273 KB
Loading

images/result-2.png

231 KB
Loading

0 commit comments

Comments
 (0)