mask maker .ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "id": "99cc7d82",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os \n",
    "import cv2 as cv \n",
    "import numpy as np \n",
    "\n",
    "import pandas as pd \n",
    "import json \n",
    "import re \n",
    "import matplotlib.pyplot as plt "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 31,
   "id": "fdd880f3",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "['ellipse', (163, 138), (9, 9), 0.165, 0, 360]\n"
     ]
    }
   ],
   "source": [
    "def reshape_poly_cordination(lst_1,lst_2):\n",
    "    new_lst=[]\n",
    "    for x,y in zip(lst_1,lst_2):\n",
    "        new_lst.append((x,y))\n",
    "    return new_lst\n",
    "#---------------------------------------------------------\n",
    "def reshape_json():\n",
    "    all_json='annotation_data_1/ALL/annotations_all.json' \n",
    "    with open(all_json,'r') as fp :\n",
    "        dict_json=json.load(fp)\n",
    "    my_list=list(dict_json.keys())\n",
    "    new_dict={}\n",
    "    for num in range(len(my_list)):\n",
    "        name=dict_json[my_list[num]]['regions'][0]['shape_attributes']['name']\n",
    "        if name== 'polygon':\n",
    "            \n",
    "            all_points_x=dict_json[my_list[num]]['regions'][0]['shape_attributes']['all_points_x']\n",
    "\n",
    "            all_points_y=dict_json[my_list[num]]['regions'][0]['shape_attributes']['all_points_y']\n",
    "            final_list=reshape_poly_cordination(all_points_x,all_points_y)\n",
    "\n",
    "            file_name=dict_json[my_list[num]]['filename']\n",
    "    #         print(file_name)\n",
    "            new_dict.update({file_name:[name,final_list]})\n",
    "        if name=='ellipse':\n",
    "            cx=int(dict_json[my_list[num]]['regions'][0]['shape_attributes']['cx'])\n",
    "            cy=int(dict_json[my_list[num]]['regions'][0]['shape_attributes']['cy'])\n",
    "            rx=int(dict_json[my_list[num]]['regions'][0]['shape_attributes']['rx'])\n",
    "            ry=int(dict_json[my_list[num]]['regions'][0]['shape_attributes']['ry'])\n",
    "            center_coordinations=(cx,cy)\n",
    "            axes_length=(rx,ry)\n",
    "            theta=dict_json[my_list[num]]['regions'][0]['shape_attributes']['theta']\n",
    "            start_angle = 0\n",
    "            end_angle = 360\n",
    "            file_name=dict_json[my_list[num]]['filename']\n",
    "            \n",
    "            new_dict.update({file_name:[name,center_coordinations,axes_length,theta,start_angle,end_angle]})\n",
    "        else : \n",
    "            continue\n",
    "            \n",
    "    return new_dict\n",
    "a=reshape_json()\n",
    "all_name=list(a.keys())\n",
    "\n",
    "print(a['y114.jpg'])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "id": "e62ccff5",
   "metadata": {},
   "outputs": [],
   "source": [
    "\n"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 32,
   "id": "1ee0ef6d",
   "metadata": {},
   "outputs": [],
   "source": [
    "a=reshape_json()\n",
    "\n",
    "def make_clean_data(img_name):\n",
    "#     img_name='y101.jpg'\n",
    "    \n",
    "    masks_path='annotation_data_1/MASKS'\n",
    "    all_image='annotation_data_1/ALL'\n",
    "    images_path='annotation_data_1/IMAGES'\n",
    "    data=a[img_name]\n",
    "    if data[0]=='polygon' : \n",
    "        image=cv.imread(all_image+'/'+img_name,0)\n",
    "\n",
    "        size=image.shape\n",
    "        \n",
    "        mask = np.zeros(size)\n",
    "        end_points=np.array(data[1])\n",
    "        cv.fillPoly(mask, pts=[end_points], color=(255))\n",
    "        \n",
    "\n",
    "    if data[0]=='ellipse':\n",
    "        image=cv.imread(all_image+'/'+img_name,0)\n",
    "        \n",
    "        \n",
    "        \n",
    "        size=image.shape\n",
    "        \n",
    "        mask = np.zeros(size)     \n",
    "        mask = cv.ellipse(mask, data[1], data[2],data[3], data[4], data[5], (255), -1)\n",
    "        \n",
    "    \n",
    "    \n",
    "    cv.imwrite(masks_path+'/'+img_name,mask)    \n",
    "    cv.imwrite(images_path+'/'+img_name,image)\n",
    "    return [mask,image]\n",
    "\n",
    "\n",
    "    "
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 33,
   "id": "dd2c888c",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "y101.jpg\n",
      "y114.jpg\n",
      "y193.jpg\n",
      "y194.jpg\n",
      "y195.jpg\n",
      "y196.jpg\n",
      "y259.jpg\n",
      "y261.jpg\n",
      "y271.jpg\n",
      "y325.jpg\n",
      "y337.jpg\n",
      "y338.jpg\n",
      "y339.jpg\n",
      "y340.jpg\n",
      "y345.jpg\n",
      "y346.jpg\n",
      "y350.jpg\n",
      "y355.jpg\n",
      "y360.jpg\n",
      "y361.jpg\n",
      "y366.jpg\n",
      "y367.jpg\n",
      "y369.jpg\n",
      "y370.jpg\n",
      "y372.jpg\n",
      "y373.jpg\n",
      "y386.jpg\n",
      "y387.jpg\n",
      "y388.jpg\n",
      "y389.jpg\n",
      "y390.jpg\n",
      "y393.jpg\n",
      "y394.jpg\n",
      "y395.jpg\n",
      "y396.jpg\n",
      "y397.jpg\n",
      "y402.jpg\n",
      "y403.jpg\n",
      "y404.jpg\n",
      "y405.jpg\n",
      "y406.jpg\n",
      "y407.jpg\n",
      "y410.jpg\n",
      "y414.jpg\n",
      "y420.jpg\n",
      "y428.jpg\n",
      "y430.jpg\n",
      "y432.jpg\n",
      "y434.jpg\n",
      "y443.jpg\n",
      "y444.jpg\n",
      "y445.jpg\n",
      "y448.jpg\n",
      "y449.jpg\n",
      "y450.jpg\n",
      "y454.jpg\n",
      "y458.jpg\n",
      "y462.jpg\n",
      "y467.jpg\n",
      "y472.jpg\n",
      "y473.jpg\n",
      "y480.jpg\n",
      "y487.jpg\n",
      "y503.jpg\n",
      "y504.jpg\n",
      "y505.jpg\n",
      "y507.jpg\n",
      "y508.jpg\n",
      "y512.jpg\n",
      "y513.jpg\n",
      "y514.jpg\n",
      "y516.jpg\n",
      "y517.jpg\n",
      "y523.jpg\n",
      "y524.jpg\n",
      "y527.jpg\n",
      "y528.jpg\n",
      "y529.jpg\n",
      "y530.jpg\n",
      "y531.jpg\n",
      "y532.jpg\n",
      "y533.jpg\n",
      "y536.jpg\n",
      "y537.jpg\n",
      "y538.jpg\n",
      "y542.jpg\n",
      "y543.jpg\n",
      "y546.jpg\n",
      "y547.jpg\n",
      "y549.jpg\n",
      "y558.jpg\n",
      "y561.jpg\n",
      "y562.jpg\n",
      "y563.jpg\n",
      "y564.jpg\n",
      "y566.jpg\n",
      "y568.jpg\n",
      "y571.jpg\n",
      "y573.jpg\n",
      "y579.jpg\n",
      "y581.jpg\n",
      "y582.jpg\n",
      "y583.jpg\n",
      "y584.jpg\n",
      "y588.jpg\n",
      "y589.jpg\n",
      "y590.jpg\n",
      "y591.jpg\n",
      "y592.jpg\n",
      "y594.jpg\n",
      "y595.jpg\n",
      "y596.jpg\n",
      "y597.jpg\n",
      "y598.jpg\n",
      "y599.jpg\n",
      "y600.jpg\n",
      "y602.jpg\n",
      "y603.jpg\n",
      "y604.jpg\n",
      "y605.jpg\n",
      "y606.jpg\n",
      "y607.jpg\n",
      "y610.jpg\n",
      "y611.jpg\n",
      "y612.jpg\n",
      "y614.jpg\n",
      "y615.jpg\n",
      "y616.jpg\n",
      "y617.jpg\n",
      "y620.jpg\n",
      "y621.jpg\n",
      "y623.jpg\n",
      "y627.jpg\n",
      "y628.jpg\n",
      "y630.jpg\n",
      "y631.jpg\n",
      "y632.jpg\n",
      "y634.jpg\n",
      "y635.jpg\n",
      "y637.jpg\n",
      "y639.jpg\n",
      "y640.jpg\n",
      "y641.jpg\n",
      "y648.jpg\n",
      "y649.jpg\n",
      "y650.jpg\n",
      "y651.jpg\n",
      "y655.jpg\n",
      "y656.jpg\n",
      "y658.jpg\n",
      "y660.jpg\n",
      "y661.jpg\n",
      "y662.jpg\n",
      "y664.jpg\n",
      "y667.jpg\n",
      "y668.jpg\n",
      "y669.jpg\n",
      "y670.jpg\n",
      "y673.jpg\n",
      "y677.jpg\n",
      "y679.jpg\n",
      "y687.jpg\n",
      "y689.jpg\n",
      "y696.jpg\n",
      "y697.jpg\n",
      "y698.jpg\n",
      "y702.jpg\n",
      "y703.jpg\n",
      "y706.jpg\n",
      "y707.jpg\n",
      "y708.jpg\n",
      "y709.jpg\n",
      "y710.jpg\n",
      "y711.jpg\n",
      "y714.jpg\n",
      "y715.jpg\n",
      "y716.jpg\n",
      "y717.jpg\n",
      "y719.jpg\n",
      "y720.jpg\n",
      "y722.jpg\n",
      "y725.jpg\n",
      "y731.jpg\n",
      "y732.jpg\n",
      "y734.jpg\n",
      "y735.jpg\n",
      "y736.jpg\n",
      "y737.jpg\n",
      "y738.jpg\n",
      "y739.jpg\n",
      "y740.jpg\n",
      "y744.jpg\n",
      "y745.jpg\n",
      "y746.jpg\n",
      "y747.jpg\n",
      "y748.jpg\n",
      "y749.jpg\n",
      "y750.jpg\n",
      "y753.jpg\n",
      "y755.jpg\n",
      "y756.jpg\n",
      "y757.jpg\n",
      "y759.jpg\n",
      "y760.jpg\n",
      "y761.jpg\n",
      "y764.jpg\n",
      "y765.jpg\n",
      "y766.jpg\n",
      "y771.jpg\n",
      "y772.jpg\n",
      "y773.jpg\n",
      "y778.jpg\n",
      "y779.jpg\n",
      "y780.jpg\n",
      "y782.jpg\n",
      "y784.jpg\n",
      "y785.jpg\n",
      "y792.jpg\n",
      "y793.jpg\n",
      "y794.jpg\n",
      "y796.jpg\n",
      "y797.jpg\n",
      "y798.jpg\n",
      "it is done \n"
     ]
    }
   ],
   "source": [
    "f=list(map(lambda x : make_clean_data(x),list(a.keys())))\n",
    "print('it is done ')"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.10.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}