pyhealth.tasks.COVID19CXRClassification

class pyhealth.tasks.covid19_cxr_classification.COVID19CXRClassification(code_mapping=None)

Bases: BaseTask

A task for classifying chest disease from chest X-ray images.

This task classifies chest X-ray images into different disease categories. It expects a single chest X-ray image per patient and returns the corresponding disease label.

task_name

The name of the task, set to “COVID19CXRClassification”.

Type:

str

input_schema

The input schema specifying the required input format. Contains a single key “image” with value “image”.

Type:

Dict[str, str]

output_schema

The output schema specifying the output format. Contains a single key “disease” with value “multiclass”.

Type:

Dict[str, str]

Examples

>>> from pyhealth.datasets import COVID19CXRDataset
>>> from pyhealth.tasks import COVID19CXRClassification
>>> dataset = COVID19CXRDataset(root="/path/to/covid19_cxr")
>>> task = COVID19CXRClassification()
>>> samples = dataset.set_task(task)

task_name: str = 'COVID19CXRClassification'

input_schema: Dict = {'image': ('image', {'mode': 'RGB'})}

output_schema: Dict[str, str] = {'disease': 'multiclass'}

pre_filter(df)

Return type:

LazyFrame