Skip to content

Latest commit

 

History

History
48 lines (43 loc) · 2.48 KB

README.md

File metadata and controls

48 lines (43 loc) · 2.48 KB

ML-Cohort-

Task 1

  • Kaggle dataset https://www.kaggle.com/mbkinaci/fruit-images-for-object-detection
  • A function in the jupyter notebook do the following tasks:
    • Input:
      • Takes the path of the folder as the input
    • Process:
      • Loads all the images and convert them to an array
      • Convert all the images to grayscale images
    • Output:
      • Number of images
      • Format type of the images
      • Saving all the grayscale images to another folder
  • Example Input apple_1 Output apple_1

Task 2

  1. Data https://drive.google.com/drive/folders/1KZ5sVLpEMqt4I4Yj3Sg8BAVJQ-5h9Nyw?usp=sharing
  2. A function in python script do the following: i. Inputs: Input the folder name and csv filename through command line ii. To-Do:
    1. Read all the images
    2. Read the csv file, and get the image label names and coordinates
    3. Draws the bounding boxes for every image, using the corresponding image’s coordinates (Use different colour and thickness for the bounding boxes)
    4. Added labels for every image (Use different font, font colour, font, size)
    5. Created another folder and saved all the images to the folder
  3. Example
  • Input cats_000
  • Output cats_000 jpg

Task 3

AIM: To create a neural network model of the highest accuracy, with the least number of parameters (can read about parameters - https://towardsdatascience.com/model-parameters-and-hyperparameters-in-machine-learning-what-is-the-difference-702d30970f6 )

  1. Download the dataset - https://drive.google.com/drive/folders/1AlztDRzHhuavHPb77o9ZDHGIp6za77AR?usp=sharing
  2. Created all the pre-processing steps similar to the one from the above notebooks
  3. Created a Neural network model using Keras and train the model on the training dataset
  4. Tested the model on the test dataset

Model architecture

image

Model Accuracy

image