MEDISEG

<b>Dataset Overview</b>This repository contains two related datasets for pill recognition research: MEDISEG and MEDISEG-Deploy.MEDISEG (MEDication Image SEGmentation) is a high-quality, real-world dataset designed for the development and evaluation of pill recognition models. It contains two subsets:MEDISEG (3-Pills): A controlled dataset featuring three pill types with subtle differences in shape and color.MEDISEG (32-Pills): A more diverse dataset containing 32 distinct pill classes, reflecting real-world challenges such as occlusions, varied lighting conditions, and multiple medications in a single frame.MEDISEG-Deploy is a deployment-oriented few-shot evaluation subset designed to support realistic pill recognition under visual domain shift. It includes cluttered multi-pill scenes with overlapping pills, varied backgrounds, pill boxes, and reflective surfaces, and provides fixed support and query partitions for few-shot adaptation under 1-shot, 5-shot, and 10-shot settings.Each dataset includes COCO-format annotations with instance segmentation masks, bounding boxes, and class labels.<b>Dataset Structure</b>The dataset is organized as follows:<br><br>MEDISEG/│── LICENSE<br>│── metadata.csv│── 3pills/<br>│ ├── annotations.json│ ├── images/<br>│ │ ├── image1.jpg│ │ ├── image2.jpg<br>│── 32pills/│ ├── annotations.json<br>│ ├── images/│ │ ├── image1.jpg<br>│ │ ├── image2.jpg<br>MEDISEG-Deploy/<br>│── LICENSE<br>│── 1-shot/<br>│ ├── annotations.json│ ├── images/<br>│ │ ├── image1.jpg│ │ ├── image2.jpg<br>│── 5-shot/│ ├── annotations.json<br>│ ├── images/│ │ ├── image1.jpg<br>│ │ ├── image2.jpg<br>│── 10-shot/<br>│ ├── annotations.json│ ├── images/<br>│ │ ├── image1.jpg│ │ ├── image2.jpg<br>│── query/│ ├── annotations.json<br>│ ├── images/│ │ ├── image1.jpg<br>│ │ ├── image2.jpg<br>LICENSE: The CC BY 4.0 license under which the dataset is distributed.metadata.csv: Supplementary drug information, including registration numbers, brand names, active ingredients, regulatory classifications, and official URLs.annotations.json: COCO-format annotation files providing segmentation masks, bounding boxes, and class labels.images/: High-resolution JPG images of medications.<b>Acknowledgements</b>If you use this dataset, please cite the corresponding publication:For MEDISEG (3-Pills and 32-Pills): <br>bibtex<br>@article{MEDISEG2026,<br>title = {A dataset of medication images with instance segmentation masks for preventing adverse drug events},<br>author = {Chu, Wai Ip and Hirani, Shashi and Tarroni, Giacomo and Li, Ling},<br>journal={arXiv preprint arXiv:2603.10825},<br>year = {2026},<br>doi={10.48550/arXiv.2603.10825}<br>}For MEDISEG-Deploy: <br>bibtex<br>@article{MEDISEGFewShot2026,<br>title = {Evaluating Few-Shot Pill Recognition Under Visual Domain Shift},<br>author = {Chu, Wai Ip and Tarroni, Giacomo and Li, Ling},<br>journal={arXiv preprint arXiv:2603.10833},<br>year = {2026},<br>doi={10.48550/arXiv.2603.10833}<br>}For work using both the core MEDISEG dataset and MEDISEG-Deploy: please cite both papers.<br>