FSCOCO

FSCOCO is a SketchDataset loader for the FS-COCO dataset, where vector sketches are stored as .npy arrays. Each array encodes a sketch as polyline points with pen-state flags and is converted into a Sketch composed of Paths and cubic Bézier Curves.

Source: datasets/fscoco.py

Data Format

Each sample is a NumPy array with shape (N, 3):

  • x: x-coordinate (float)

  • y: y-coordinate (float)

  • flag: pen state (int)

Flag convention used by this loader:

  • 0: pen-down / continue current stroke

  • 1: pen-up (end current stroke; continue parsing)

  • 2: end-of-drawing (end current stroke; stop parsing)

The loader groups points into strokes and converts each consecutive point pair into a cubic Bézier curve.

Directory Layout

After download and extraction, the dataset is expected under:

<root>/
  fscoco/
    vector_sketches/
      1/
        *.npy
      2/
        *.npy
      ...
      100/
        *.npy
  .metadata.parquet

  • Shards are expected to be numbered 1..100.

  • Each shard contains multiple .npy sketch files.

Code

from sketchkit.datasets import FSCOCO

ds = FSCOCO(
    root="path/to/cache_dir",
    load_all=True,
    cislab_source=True,
)

sketch = ds[0]
print(sketch.width, sketch.height)
print(sketch.path_num, sketch.curve_num)

Arguments

  • root: Root directory used for caching and extraction. FS-COCO data is placed under <root>/fscoco/.

  • load_all: If True, preload all .npy arrays into memory to avoid disk I/O during iteration.

  • cislab_source: Selects the download source. If True, the dataset is downloaded from the CISLAB CDN mirror; otherwise it is downloaded from the official host.

    • CISLAB mirror: https://cislab.hkust-gz.edu.cn/projects/sketchkit/datasets/FSCOCO/fscoco.tar.gz

    • Official host: http://cvssp.org/data/fscoco/fscoco.tar.gz