Create README.md
Browse files
README.md
ADDED
@@ -0,0 +1,83 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
---
|
4 |
+
|
5 |
+
# ArtifyAI - COCO Dataset Downloader and Organizer
|
6 |
+
|
7 |
+
Welcome to ArtifyAI, a project designed to download images from the COCO dataset and organize them into your Google Drive for further use in machine learning and computer vision projects.
|
8 |
+
|
9 |
+
## Introduction
|
10 |
+
|
11 |
+
This project automates the process of downloading a subset of images from the COCO dataset and storing them directly in Google Drive. It's aimed at researchers and developers who need a quick way to set up their dataset for training image-based models. The notebook is structured to be easy to follow, even for non-technical users.
|
12 |
+
|
13 |
+
## Requirements
|
14 |
+
|
15 |
+
To use this notebook, you need:
|
16 |
+
|
17 |
+
- Google Colab (optional, but recommended for ease of use)
|
18 |
+
- Google Drive (to store the dataset)
|
19 |
+
- Basic knowledge of Python (optional)
|
20 |
+
- Internet connection (for downloading images)
|
21 |
+
- Libraries:
|
22 |
+
- `pycocotools`
|
23 |
+
- `requests`
|
24 |
+
- `tqdm`
|
25 |
+
- `matplotlib`
|
26 |
+
- `numpy`
|
27 |
+
- `shutil`
|
28 |
+
|
29 |
+
## Installation
|
30 |
+
|
31 |
+
1. **Clone the repository (optional)**:
|
32 |
+
```bash
|
33 |
+
git clone https://github.com/your-repo/ArtifyAI.git
|
34 |
+
cd ArtifyAI
|
35 |
+
```
|
36 |
+
|
37 |
+
2. **Open the notebook**:
|
38 |
+
- You can directly upload the notebook (`ArtifyAI_v1_0.ipynb`) to your Google Colab environment or use it locally in your Python environment.
|
39 |
+
|
40 |
+
3. **Install dependencies**:
|
41 |
+
If you're running this locally, make sure to install the required libraries:
|
42 |
+
```bash
|
43 |
+
pip install pycocotools tqdm matplotlib numpy requests
|
44 |
+
```
|
45 |
+
|
46 |
+
## Running the Project
|
47 |
+
|
48 |
+
Once all dependencies are installed, follow these steps:
|
49 |
+
|
50 |
+
1. **Open Google Colab**:
|
51 |
+
- If you're using Google Colab, upload the notebook to your Colab environment.
|
52 |
+
|
53 |
+
2. **Mount Google Drive**:
|
54 |
+
- The notebook will prompt you to mount your Google Drive. Ensure your account is connected so that the images are downloaded and saved to a designated folder in your drive.
|
55 |
+
|
56 |
+
3. **Run the Code**:
|
57 |
+
- Execute the cells sequentially to start downloading the COCO dataset images. The notebook uses `pycocotools` to fetch image URLs from the COCO API and downloads them using `requests`. Progress is tracked with `tqdm`.
|
58 |
+
|
59 |
+
4. **Transfer Images to Google Drive**:
|
60 |
+
- After downloading, the images will automatically be moved to a folder in your Google Drive (`MyDrive/coco_dataset`).
|
61 |
+
|
62 |
+
## Using Google Colab
|
63 |
+
|
64 |
+
For non-technical users, we recommend using [Google Colab](https://colab.research.google.com/). It provides a cloud-based environment where you can run the notebook without installing Python or any dependencies on your local machine.
|
65 |
+
|
66 |
+
1. **Upload the Notebook**:
|
67 |
+
- Simply drag and drop the `.ipynb` file into Colab.
|
68 |
+
|
69 |
+
2. **Mount Google Drive**:
|
70 |
+
- The notebook includes a step to mount Google Drive for file storage. Follow the on-screen instructions to authorize access.
|
71 |
+
|
72 |
+
3. **Run the Notebook**:
|
73 |
+
- Execute each cell in the notebook by clicking the play button next to each cell. Ensure all code cells are run in order.
|
74 |
+
|
75 |
+
## Moving Files to Google Drive
|
76 |
+
|
77 |
+
After downloading the images, the notebook will move them to a specified folder in your Google Drive, making it easy for you to access them later.
|
78 |
+
|
79 |
+
- The default folder is `/content/drive/MyDrive/coco_dataset`, but you can modify the path if needed.
|
80 |
+
|
81 |
+
## License
|
82 |
+
|
83 |
+
This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.
|