Edit model card
keremberke/yolov8n-chest-xray-classification

Supported Labels

['NORMAL', 'PNEUMONIA']

How to use

pip install ultralyticsplus==0.0.23 ultralytics==8.0.21
  • Load model and perform prediction:
from ultralyticsplus import YOLO, postprocess_classify_output

# load model
model = YOLO('keremberke/yolov8n-chest-xray-classification')

# set model parameters
model.overrides['conf'] = 0.25  # model confidence threshold

# set image
image = 'https://github.com/ultralytics/yolov5/raw/master/data/images/zidane.jpg'

# perform inference
results = model.predict(image)

# observe results
print(results[0].probs) # [0.1, 0.2, 0.3, 0.4]
processed_result = postprocess_classify_output(model, result=results[0])
print(processed_result) # {"cat": 0.4, "dog": 0.6}

More models available at: awesome-yolov8-models

Downloads last month
2,866
Inference API
Drag image file here or click to browse from your device
Inference API (serverless) has been turned off for this model.

Dataset used to train keremberke/yolov8n-chest-xray-classification

Evaluation results