Skip to content

YunYang1994/yymnist

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Introduction

A lot of wonderful datasets are now available online, such as COCO or Imagenet. which is challenging the limits of computer vision. But it's not easy for us to do some small experiments with such a large number of images to quickly test the validity of algorithmn. For this reason, I created a small dataset named "yymnist" to do both classification and object detection.

classification object detection
image image

Installation

Clone this repo and install some dependencies

$ pip install opencv-python
$ git clone https://github.com/YunYang1994/yymnist.git

Classification

If you want to use this dataset for classification, just use mnist folder. Each image in the folder starts with a name of its label. For example:

mnist/train/0_32970.pgm -> 0
mnist/train/5_12156.pgm -> 5

Objection Detection

$ python yymnist/make_data.py
$ python yymnist/show_image.py # [option]

you will defaultly get 1000 pictures in the folder ./yymnist/Images and a label.txt in the ./yymnist/. As for label.txt

# image_path x_min, y_min, x_max, y_max, class_id  x_min, y_min ,..., class_id 

About

a mnist dataset for object detection

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages