dataflow-cv 0.2.0__tar.gz

dataflow_cv-0.2.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 zjykzj
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

dataflow_cv-0.2.0/PKG-INFO

@@ -0,0 +1,202 @@
+Metadata-Version: 2.4
+Name: dataflow-cv
+Version: 0.2.0
+Summary: A data processing library for computer vision datasets
+Home-page: https://github.com/zjykzj/DataFlow-CV
+Author: DataFlow Team
+License: MIT
+Keywords: computer-vision,dataset,conversion,visualization,coco,yolo,labelme
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: POSIX :: Linux
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.8
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Topic :: Scientific/Engineering :: Image Processing
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=2.0.0
+Requires-Dist: opencv-python>=4.8.0
+Requires-Dist: Pillow>=10.0.0
+Requires-Dist: click>=8.1.0
+Dynamic: home-page
+Dynamic: license-file
+Dynamic: requires-python
+
+# DataFlow-CV
+
+> **Where Vibe Coding meets CV data.** 🌊
+> Convert & visualize datasets. Built with the flow of Claude Code.
+
+![Python Version](https://img.shields.io/badge/python-3.8%20|%203.9%20|%203.10%20|%203.11%20|%203.12-blue)
+![License](https://img.shields.io/badge/license-MIT-green)
+![Version](https://img.shields.io/badge/version-0.2.0-orange)
+![Development Status](https://img.shields.io/badge/status-alpha-yellow)
+
+A data processing library for computer vision datasets, focusing on format conversion and visualization between LabelMe, COCO, and YOLO formats. Provides both a CLI and Python API.
+
+## Table of Contents
+
+- [DataFlow-CV](#dataflow-cv)
+  - [Table of Contents](#table-of-contents)
+  - [Project Structure](#project-structure)
+  - [Requirements](#requirements)
+  - [License](#license)
+
+## Project Structure
+
+```
+dataflow/
+├── __init__.py              # Package exports and convenience functions
+├── cli.py                   # Command-line interface
+├── config.py                # Configuration management
+├── convert/                 # Format conversion module
+│   ├── __init__.py
+│   ├── base.py             # Converter base class
+│   ├── coco_to_yolo.py     # COCO to YOLO converter
+│   └── yolo_to_coco.py     # YOLO to COCO converter
+tests/
+├── __init__.py
+├── convert/                # Conversion tests
+│   ├── __init__.py
+│   ├── test_coco_to_yolo.py
+│   └── test_yolo_to_coco.py
+├── run_tests.py           # Test runner
+samples/
+├── __init__.py
+├── cli/                   # CLI usage examples
+│   ├── __init__.py
+│   └── convert/
+│       ├── cli_coco_to_yolo.py
+│       └── cli_yolo_to_coco.py
+└── api/                   # Python API examples
+    ├── __init__.py
+    └── convert/
+        ├── api_coco_to_yolo.py
+        └── api_yolo_to_coco.py
+```
+
+## Requirements
+
+### Core Dependencies
+- Python 3.8 or higher
+- Linux environment (POSIX compatible, assumes POSIX paths)
+- `click` >= 8.1.0 – CLI framework
+- `numpy` >= 2.0.0 – numerical operations
+- `opencv-python` >= 4.8.0 – image processing (optional, used for some image operations)
+- `Pillow` >= 10.0.0 – image reading (optional, used for reading image dimensions)
+
+## Quick Start
+
+### Installation
+
+```bash
+# Install
+pip install .
+```
+
+### Command Line Usage
+
+Global options: `--verbose` (`-v`) for progress output, `--overwrite` to replace existing files.
+
+```bash
+# COCO to YOLO conversion
+dataflow convert coco2yolo annotations.json output_dir/
+
+# YOLO to COCO conversion
+dataflow convert yolo2coco images/ labels/ classes.names output.json
+
+# Show configuration
+dataflow config
+
+# Get help
+dataflow --help
+dataflow convert coco2yolo --help
+```
+
+See the [CLI Reference](#cli-reference) below for detailed usage.
+
+### Python API Usage
+
+```python
+import dataflow
+
+# COCO to YOLO
+result = dataflow.coco_to_yolo("annotations.json", "output_dir")
+print(f"Processed {result['images_processed']} images")
+
+# YOLO to COCO
+result = dataflow.yolo_to_coco("images/", "labels/", "classes.names", "output.json")
+print(f"Generated {result['annotations_processed']} annotations")
+```
+
+### CLI Reference
+
+The CLI follows a hierarchical structure: `dataflow <main‑task> <sub‑task> [arguments]`. Global options can be placed before the main task.
+
+#### Global Options
+- `--verbose`, `-v`: Enable verbose output (progress information)
+- `--overwrite`: Overwrite existing files
+
+#### Conversion Commands
+
+**COCO to YOLO**
+```bash
+dataflow convert coco2yolo COCO_JSON_PATH OUTPUT_DIR [--segmentation]
+```
+- `COCO_JSON_PATH`: Path to COCO JSON annotation file
+- `OUTPUT_DIR`: Directory where `labels/` and `class.names` will be created
+- `--segmentation`, `-s`: Handle segmentation annotations (polygon format)
+
+**YOLO to COCO**
+```bash
+dataflow convert yolo2coco IMAGE_DIR YOLO_LABELS_DIR YOLO_CLASS_PATH COCO_JSON_PATH
+```
+- `IMAGE_DIR`: Directory containing image files
+- `YOLO_LABELS_DIR`: Directory containing YOLO label files (`.txt`)
+- `YOLO_CLASS_PATH`: Path to YOLO class names file (e.g., `class.names`)
+- `COCO_JSON_PATH`: Path to save COCO JSON file
+
+#### Configuration Command
+```bash
+dataflow config
+```
+Shows the current configuration (file extensions, default values, CLI context).
+
+#### Getting Help
+```bash
+dataflow --help
+dataflow convert --help
+dataflow convert coco2yolo --help
+dataflow convert yolo2coco --help
+```
+
+### Running Tests
+
+```bash
+# Run all tests
+python tests/run_tests.py
+
+# Run specific test
+python tests/run_tests.py --test TestCocoToYoloConverter
+
+# With verbose output
+python tests/run_tests.py -v
+```
+
+### Examples
+
+Check the `samples/` directory for detailed usage examples:
+
+- `samples/cli/convert/` - CLI usage examples
+- `samples/api/convert/` - Python API examples
+
+## License
+
+[MIT License](LICENSE) © 2026 zjykzj

dataflow_cv-0.2.0/README.md

@@ -0,0 +1,171 @@
+# DataFlow-CV
+
+> **Where Vibe Coding meets CV data.** 🌊
+> Convert & visualize datasets. Built with the flow of Claude Code.
+
+![Python Version](https://img.shields.io/badge/python-3.8%20|%203.9%20|%203.10%20|%203.11%20|%203.12-blue)
+![License](https://img.shields.io/badge/license-MIT-green)
+![Version](https://img.shields.io/badge/version-0.2.0-orange)
+![Development Status](https://img.shields.io/badge/status-alpha-yellow)
+
+A data processing library for computer vision datasets, focusing on format conversion and visualization between LabelMe, COCO, and YOLO formats. Provides both a CLI and Python API.
+
+## Table of Contents
+
+- [DataFlow-CV](#dataflow-cv)
+  - [Table of Contents](#table-of-contents)
+  - [Project Structure](#project-structure)
+  - [Requirements](#requirements)
+  - [License](#license)
+
+## Project Structure
+
+```
+dataflow/
+├── __init__.py              # Package exports and convenience functions
+├── cli.py                   # Command-line interface
+├── config.py                # Configuration management
+├── convert/                 # Format conversion module
+│   ├── __init__.py
+│   ├── base.py             # Converter base class
+│   ├── coco_to_yolo.py     # COCO to YOLO converter
+│   └── yolo_to_coco.py     # YOLO to COCO converter
+tests/
+├── __init__.py
+├── convert/                # Conversion tests
+│   ├── __init__.py
+│   ├── test_coco_to_yolo.py
+│   └── test_yolo_to_coco.py
+├── run_tests.py           # Test runner
+samples/
+├── __init__.py
+├── cli/                   # CLI usage examples
+│   ├── __init__.py
+│   └── convert/
+│       ├── cli_coco_to_yolo.py
+│       └── cli_yolo_to_coco.py
+└── api/                   # Python API examples
+    ├── __init__.py
+    └── convert/
+        ├── api_coco_to_yolo.py
+        └── api_yolo_to_coco.py
+```
+
+## Requirements
+
+### Core Dependencies
+- Python 3.8 or higher
+- Linux environment (POSIX compatible, assumes POSIX paths)
+- `click` >= 8.1.0 – CLI framework
+- `numpy` >= 2.0.0 – numerical operations
+- `opencv-python` >= 4.8.0 – image processing (optional, used for some image operations)
+- `Pillow` >= 10.0.0 – image reading (optional, used for reading image dimensions)
+
+## Quick Start
+
+### Installation
+
+```bash
+# Install
+pip install .
+```
+
+### Command Line Usage
+
+Global options: `--verbose` (`-v`) for progress output, `--overwrite` to replace existing files.
+
+```bash
+# COCO to YOLO conversion
+dataflow convert coco2yolo annotations.json output_dir/
+
+# YOLO to COCO conversion
+dataflow convert yolo2coco images/ labels/ classes.names output.json
+
+# Show configuration
+dataflow config
+
+# Get help
+dataflow --help
+dataflow convert coco2yolo --help
+```
+
+See the [CLI Reference](#cli-reference) below for detailed usage.
+
+### Python API Usage
+
+```python
+import dataflow
+
+# COCO to YOLO
+result = dataflow.coco_to_yolo("annotations.json", "output_dir")
+print(f"Processed {result['images_processed']} images")
+
+# YOLO to COCO
+result = dataflow.yolo_to_coco("images/", "labels/", "classes.names", "output.json")
+print(f"Generated {result['annotations_processed']} annotations")
+```
+
+### CLI Reference
+
+The CLI follows a hierarchical structure: `dataflow <main‑task> <sub‑task> [arguments]`. Global options can be placed before the main task.
+
+#### Global Options
+- `--verbose`, `-v`: Enable verbose output (progress information)
+- `--overwrite`: Overwrite existing files
+
+#### Conversion Commands
+
+**COCO to YOLO**
+```bash
+dataflow convert coco2yolo COCO_JSON_PATH OUTPUT_DIR [--segmentation]
+```
+- `COCO_JSON_PATH`: Path to COCO JSON annotation file
+- `OUTPUT_DIR`: Directory where `labels/` and `class.names` will be created
+- `--segmentation`, `-s`: Handle segmentation annotations (polygon format)
+
+**YOLO to COCO**
+```bash
+dataflow convert yolo2coco IMAGE_DIR YOLO_LABELS_DIR YOLO_CLASS_PATH COCO_JSON_PATH
+```
+- `IMAGE_DIR`: Directory containing image files
+- `YOLO_LABELS_DIR`: Directory containing YOLO label files (`.txt`)
+- `YOLO_CLASS_PATH`: Path to YOLO class names file (e.g., `class.names`)
+- `COCO_JSON_PATH`: Path to save COCO JSON file
+
+#### Configuration Command
+```bash
+dataflow config
+```
+Shows the current configuration (file extensions, default values, CLI context).
+
+#### Getting Help
+```bash
+dataflow --help
+dataflow convert --help
+dataflow convert coco2yolo --help
+dataflow convert yolo2coco --help
+```
+
+### Running Tests
+
+```bash
+# Run all tests
+python tests/run_tests.py
+
+# Run specific test
+python tests/run_tests.py --test TestCocoToYoloConverter
+
+# With verbose output
+python tests/run_tests.py -v
+```
+
+### Examples
+
+Check the `samples/` directory for detailed usage examples:
+
+- `samples/cli/convert/` - CLI usage examples
+- `samples/api/convert/` - Python API examples
+
+## License
+
+[MIT License](LICENSE) © 2026 zjykzj

dataflow_cv-0.2.0/dataflow/__init__.py

@@ -0,0 +1,150 @@
+# -*- coding: utf-8 -*-
+
+"""
+@Time    : 2026/3/8 20:41
+@File    : __init__.py
+@Author  : zj
+@Description: DataFlow-CV: A data processing library for computer vision datasets
+"""
+
+__version__ = "0.2.0"
+__author__ = "DataFlow Team"
+__description__ = "A data processing library for computer vision datasets"
+
+# Configuration
+from .config import Config
+
+# Converters
+from .convert.coco_to_yolo import CocoToYoloConverter
+from .convert.yolo_to_coco import YoloToCocoConverter
+
+# Visualizers
+from .visualize.yolo import YoloVisualizer
+from .visualize.coco import CocoVisualizer
+
+# Base classes
+from .convert.base import BaseConverter
+
+# CLI
+from .cli import cli, main
+
+# Convenience functions for common conversions
+def coco_to_yolo(coco_json_path: str, output_dir: str, **kwargs):
+    """
+    Convert COCO JSON to YOLO format.
+
+    Args:
+        coco_json_path: Path to COCO JSON file
+        output_dir: Output directory where labels/ and class.names will be created
+        **kwargs: Additional options passed to CocoToYoloConverter.convert()
+
+    Returns:
+        Dictionary with conversion statistics
+    """
+    converter = CocoToYoloConverter()
+    return converter.convert(coco_json_path, output_dir, **kwargs)
+
+
+def yolo_to_coco(
+    image_dir: str,
+    yolo_labels_dir: str,
+    yolo_class_path: str,
+    coco_json_path: str,
+    **kwargs
+):
+    """
+    Convert YOLO format to COCO JSON.
+
+    Args:
+        image_dir: Directory containing image files
+        yolo_labels_dir: Directory containing YOLO label files
+        yolo_class_path: Path to YOLO class names file
+        coco_json_path: Path to save COCO JSON file
+        **kwargs: Additional options passed to YoloToCocoConverter.convert()
+
+    Returns:
+        Dictionary with conversion statistics
+    """
+    converter = YoloToCocoConverter()
+    return converter.convert(
+        image_dir, yolo_labels_dir, yolo_class_path, coco_json_path, **kwargs
+    )
+
+
+# Convenience functions for visualization
+def visualize_yolo(
+    image_dir: str,
+    label_dir: str,
+    class_path: str,
+    save_dir: str = None,
+    **kwargs
+):
+    """
+    Visualize YOLO format annotations.
+
+    Args:
+        image_dir: Directory containing image files
+        label_dir: Directory containing YOLO label files
+        class_path: Path to class names file (e.g., class.names)
+        save_dir: Directory to save visualized images (optional)
+        **kwargs: Additional options passed to YoloVisualizer.visualize()
+
+    Returns:
+        Dictionary with visualization statistics
+    """
+    visualizer = YoloVisualizer(**kwargs)
+    return visualizer.visualize(image_dir, label_dir, class_path, save_dir)
+
+
+def visualize_coco(
+    image_dir: str,
+    annotation_json: str,
+    save_dir: str = None,
+    **kwargs
+):
+    """
+    Visualize COCO format annotations.
+
+    Args:
+        image_dir: Directory containing image files
+        annotation_json: Path to COCO JSON annotation file
+        save_dir: Directory to save visualized images (optional)
+        **kwargs: Additional options passed to CocoVisualizer.visualize()
+
+    Returns:
+        Dictionary with visualization statistics
+    """
+    visualizer = CocoVisualizer(**kwargs)
+    return visualizer.visualize(image_dir, annotation_json, save_dir)
+
+
+__all__ = [
+    # Configuration
+    "Config",
+
+    # Converters
+    "CocoToYoloConverter",
+    "YoloToCocoConverter",
+
+    # Visualizers
+    "YoloVisualizer",
+    "CocoVisualizer",
+
+    # Base classes
+    "BaseConverter",
+
+    # CLI
+    "cli",
+    "main",
+
+    # Convenience functions
+    "coco_to_yolo",
+    "yolo_to_coco",
+    "visualize_yolo",
+    "visualize_coco",
+
+    # Metadata
+    "__version__",
+    "__author__",
+    "__description__",
+]