This model is based on the research described in the paper titled "Enhancing Cervical Cancer Cytology Screening via Artificial Intelligence Innovation". The research discusses how the application of advanced AI techniques can significantly improve the accuracy and efficiency of cervical cancer screening, offering a more scalable and cost-effective solution compared to traditional methods. Specifically, this model focuses on classifying tile images from LBC cytology specimens at low magnification (x10), distinguishing between normal and abnormal categories, which is a departure from the high-magnification or single-cell focused approaches commonly used in cytology.
This model is not intended to be used in isolation. To fully utilize its capabilities and implement the techniques developed, please refer to the accompanying code available on our
GitHub repository. The code provides necessary details on how to effectively use the model in your applications.
For full documentation, example scripts, and more details, visit our GitHub repository.
1from PIL import Image
2from transformers import CLIPModel, CLIPProcessor
3
4device = "cuda:0" if torch.cuda.is_available() else "cpu"
5
6model = CLIPModel.from_pretrained(kuri54/GynAIe-preview-B16-5k)
7processor = CLIPProcessor.from_pretrained(kuri54/GynAIe-preview-B16-5k)
8
9image = Image.open(path/to/image)
10
11# normal or abnormal
12text = ['a image of a normal', 'a image of a anomaly']
13
14inputs = processor(text=text, images=image, return_tensors='pt', padding=True).to(device)
15
16outputs = model(**inputs)
17probs = outputs.logits_per_image.softmax(dim=1).cpu().detach().numpy()
18predicted_class_idx = probs.argmax(-1).item()
19
20print('Class:', labels[predicted_class_idx])
21print('Score:', probs)
22
