KeerthiVM commited on
Commit
fd5144d
·
1 Parent(s): 87c2216
Files changed (1) hide show
  1. SkinCancerDiagnosis.py +2 -1
SkinCancerDiagnosis.py CHANGED
@@ -131,6 +131,7 @@ class SkinDiseaseClassifier:
131
  self.base_models = None
132
  self.meta_model = None
133
  self.resnet_feature_extractor = None
 
134
 
135
  # Image transformations
136
  self.transform = transforms.Compose([
@@ -291,7 +292,7 @@ class SkinDiseaseClassifier:
291
  with torch.no_grad():
292
  output_multi = self.skincon_model(image_tensor)
293
  probs_multi = torch.sigmoid(output_multi).squeeze().numpy()
294
- # print(f"Probabilities : {probs_multi}")
295
  threshold = 0.5
296
  predicted_labels_multi = [self.multilabel_class_names[i] for i, p in enumerate(probs_multi) if p > threshold]
297
  print("Predicted labels multi : ",predicted_labels_multi)
 
131
  self.base_models = None
132
  self.meta_model = None
133
  self.resnet_feature_extractor = None
134
+ self.skincon_model = None
135
 
136
  # Image transformations
137
  self.transform = transforms.Compose([
 
292
  with torch.no_grad():
293
  output_multi = self.skincon_model(image_tensor)
294
  probs_multi = torch.sigmoid(output_multi).squeeze().numpy()
295
+ print(f"Probabilities : {probs_multi}")
296
  threshold = 0.5
297
  predicted_labels_multi = [self.multilabel_class_names[i] for i, p in enumerate(probs_multi) if p > threshold]
298
  print("Predicted labels multi : ",predicted_labels_multi)