ShapeSet 0.1.1

This package helps generate datasets of simple shapes like squares, circles, triangles, etc. for computer vision tasks.

ShapeSet

SHapeSet is a package designed to simplify and streamline your workflow for [briefly describe the main purpose or functionality of your package, e.g., "data processing", "API integration", etc.].

Features

• Easy to use
• Lightweight and efficient
• Flexible data generation
• Scalable in data size

Installation

pip install ShapeSet

Usage

from ShapeSet import dataset

# Example usage
dataset.generate(
        class_names,
        num_train,
        num_val,
        image_size,
        output_dir,
        seed
)

Note

Therefore, the minimum allowable image size is 80x80 pixels.
If you use a smaller size, you may get a ValueError: empty range for randrange().

Recommendation:
Use image_size >= 80 in your tests and dataset generation.

Contributing

Contributions are welcome! Please open issues or submit pull requests.

License

This project is licensed under the MIT License.