Skip to content
Snippets Groups Projects
Commit e26fe458 authored by Daniel CARRON's avatar Daniel CARRON :b:
Browse files

Removed ununsed colors list

parent 2bc861d4
No related branches found
No related tags found
1 merge request!4Moved code to lightning
......@@ -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.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment