From e26fe4581eef7fd86b77a71dcbe292a61bf1c50a Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Fri, 12 May 2023 12:27:53 +0200
Subject: [PATCH] Removed ununsed colors list
---
src/ptbench/models/pasa.py | 17 -----------------
1 file changed, 17 deletions(-)
diff --git a/src/ptbench/models/pasa.py b/src/ptbench/models/pasa.py
index d4e5b2a8..78dff8e3 100644
--- a/src/ptbench/models/pasa.py
+++ b/src/ptbench/models/pasa.py
@@ -9,23 +9,6 @@ import torch.nn.functional as F
from .normalizer import TorchVisionNormalizer
-colors = [
- [(47, 79, 79), "Cardiomegaly"],
- [(255, 0, 0), "Emphysema"],
- [(0, 128, 0), "Pleural effusion"],
- [(0, 0, 128), "Hernia"],
- [(255, 84, 0), "Infiltration"],
- [(222, 184, 135), "Mass"],
- [(0, 255, 0), "Nodule"],
- [(0, 191, 255), "Atelectasis"],
- [(0, 0, 255), "Pneumothorax"],
- [(255, 0, 255), "Pleural thickening"],
- [(255, 255, 0), "Pneumonia"],
- [(126, 0, 255), "Fibrosis"],
- [(255, 20, 147), "Edema"],
- [(0, 255, 180), "Consolidation"],
-]
-
class PASA(pl.LightningModule):
"""PASA module.
--
GitLab