Skip to content

Commit

Permalink
dev version 190623_2000
Browse files Browse the repository at this point in the history
  • Loading branch information
@lbborkowski
lbborkowski committed Jun 24, 2019
1 parent e590d95 commit 24518dd
Showing 1 changed file with 136 additions and 91 deletions.
227 changes: 136 additions & 91 deletions WindTurbineDetector_dev.ipynb
View file
Original file line number Diff line number Diff line change
Expand Up @@ -90,53 +90,6 @@
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "eU99BVQ-wbaT",
"colab_type": "code",
"colab": {}
},
"source": [
"width, height = image.size\n",
"print(width)\n",
"print(height)"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "vZaiQXMSw5i-",
"colab_type": "code",
"colab": {}
},
"source": [
"box = (300, 300,\n",
" width-300,\n",
" height-300)\n",
"image.crop(box).save('test.jpg')"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "CuLu6Cl4xYxO",
"colab_type": "code",
"colab": {}
},
"source": [
"testIm=Image.open('test.jpg')\n",
"test_width, test_height = testIm.size\n",
"print(test_width)\n",
"print(test_height)"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "markdown",
"metadata": {
Expand Down Expand Up @@ -925,19 +878,6 @@
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "KiYNgvCh9Xqq",
"colab_type": "code",
"colab": {}
},
"source": [
"(len(BBsWTs),2)"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
Expand Down Expand Up @@ -968,14 +908,14 @@
"colab": {}
},
"source": [
"westEdge=[]\n",
"eastEdge=[]\n",
"northEdge=[]\n",
"southEdge=[]\n",
"#westEdge=[]\n",
"#eastEdge=[]\n",
"#northEdge=[]\n",
"#southEdge=[]\n",
"\n",
"for WTs in detectedWTs:\n",
" fid=open(WTs[0].replace('jpg','txt'))\n",
" lines=fid.readlines()\n",
"#for WTs in detectedWTs:\n",
"# fid=open(WTs[0].replace('jpg','txt'))\n",
"# lines=fid.readlines()\n",
" #print(lines[72])\n",
" #print(lines[72][33:])\n",
" #print(lines[73])\n",
Expand All @@ -990,54 +930,159 @@
" #northEdge.append(float(lines[74][34:]))\n",
" #southEdge.append(float(lines[75][34:]))\n",
" \n",
" westEdge.append(-94.8178888)\n",
" eastEdge.append(-94.7464999)\n",
" northEdge.append(41.3151166)\n",
" southEdge.append(41.2473638)\n",
" fid.close()\n",
" \n",
"print(westEdge)\n",
"print(eastEdge)\n",
"print(northEdge)\n",
"print(southEdge)"
"# westEdge.append(-94.8178888)\n",
"# eastEdge.append(-94.7464999)\n",
"# northEdge.append(41.3151166)\n",
"# southEdge.append(41.2473638)\n",
"# fid.close()\n",
"# \n",
"#print(westEdge)\n",
"#print(eastEdge)\n",
"#print(northEdge)\n",
"#print(southEdge)"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "Og7f9ir03xUo",
"id": "CDwOHgylt961",
"colab_type": "code",
"colab": {}
},
"source": [
"print(height)\n",
"print(width)\n",
"print(chipsize)\n",
"print(detectedWTs[0][2])\n",
"print(detectedWTs[0][1])\n",
"print(centerBBs[0][1])"
"#print(len(detectedWTs))\n",
"#latlongWTs=[]\n",
"#for ll in range(len(detectedWTs)):\n",
"# latlongWTs.append([westEdge[ll]+chipsize*((detectedWTs[ll][1]-1)+centerBBs[ll][0])/width*(eastEdge[ll]-westEdge[ll]), \n",
"# northEdge[ll]+chipsize*((detectedWTs[ll][2]-1)+centerBBs[ll][1])/height*(southEdge[ll]-northEdge[ll])])\n",
"# latlongWTs.append([northEdge[ll]+chipsize*((detectedWTs[ll][2]-1)+centerBBs[ll][0])/height*(southEdge[ll]-northEdge[ll]),\n",
"# westEdge[ll]+chipsize*((detectedWTs[ll][1]-1)+centerBBs[ll][1])/width*(eastEdge[ll]-westEdge[ll])])\n",
"#print(latlongWTs)"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "CDwOHgylt961",
"id": "M5QTIrHGH6Iz",
"colab_type": "code",
"colab": {}
},
"source": [
"#print(len(detectedWTs))\n",
"x1=-94.8178888\n",
"y1=41.3151166\n",
"x2=-94.7464999\n",
"y2=41.3162222\n",
"x3=-94.7446638\n",
"y3=41.2473638\n",
"x4=-94.8159777\n",
"y4=41.246261\n",
"\n",
"#ll=0\n",
"latlongWTs=[]\n",
"for ll in range(len(detectedWTs)):\n",
"# latlongWTs.append([westEdge[ll]+chipsize*((detectedWTs[ll][1]-1)+centerBBs[ll][0])/width*(eastEdge[ll]-westEdge[ll]), \n",
"# northEdge[ll]+chipsize*((detectedWTs[ll][2]-1)+centerBBs[ll][1])/height*(southEdge[ll]-northEdge[ll])])\n",
" latlongWTs.append([northEdge[ll]+chipsize*((detectedWTs[ll][2]-1)+centerBBs[ll][0])/height*(southEdge[ll]-northEdge[ll]),\n",
" westEdge[ll]+chipsize*((detectedWTs[ll][1]-1)+centerBBs[ll][1])/width*(eastEdge[ll]-westEdge[ll])])\n",
"print(latlongWTs)"
" a5=chipsize*(detectedWTs[ll][1]-1+centerBBs[ll][0])\n",
" b5=chipsize*(detectedWTs[ll][2]-1+centerBBs[ll][1])\n",
"\n",
" p_a5=a5/width\n",
" p_b5=b5/height\n",
"\n",
" x_le=x1+p_b5*(x4-x1)\n",
" x_re=x2+p_b5*(x3-x2)\n",
" x_WT=x_le+p_a5*(x_re-x_le)\n",
"\n",
" y_te=y1+p_a5*(y2-y1)\n",
" y_be=y4+p_a5*(y3-y4)\n",
" y_WT=y_te+p_b5*(y_be-y_te)\n",
" latlongWTs.append([y_WT,x_WT])\n",
"\n",
"print(latlongWTs)\n"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "XlPLn9Mtjpgn",
"colab_type": "code",
"colab": {}
},
"source": [
"for image_path in TEST_IMAGE_PATHS:\n",
" image = Image.open(image_path)\n",
" plt.figure(figsize=(width/500,height/500))\n",
" plt.axis('off')\n",
" for ll in range(len(detectedWTs)):\n",
" if detectedWTs[ll][0]==image_path:\n",
" x_plot=chipsize*(detectedWTs[ll][1]-1+centerBBs[ll][0])\n",
" y_plot=chipsize*(detectedWTs[ll][2]-1+centerBBs[ll][1])\n",
" plt.plot(x_plot,y_plot,'ro')\n",
" plt.imshow(image)\n"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "f-eGboJlqZzy",
"colab_type": "code",
"colab": {}
},
"source": [
"detectedWTs[0][0]"
],
"execution_count": 0,
"outputs": []
},
{
"cell_type": "code",
"metadata": {
"id": "pc0NwxvFqst2",
"colab_type": "code",
"colab": {}
},
"source": [
"x1={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':-94.8178888,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':-94.8180499}\n",
"y1={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':41.3151166,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':42.6276193}\n",
"x2={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':-94.7464999,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':-94.7464221}\n",
"y2={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':41.3162222,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':42.6287332}\n",
"x3={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':-94.7446638,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':-94.7444999}\n",
"y3={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':41.2473638,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':42.5598722}\n",
"x4={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':-94.8159777,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':-94.8160472}\n",
"y4={'/content/wind-turbine-detector/images/samples/orig/m_4109442_se_15_1_20170709.jpg':41.246261,\n",
" '/content/wind-turbine-detector/images/samples/orig/m_4209426_ne_15_1_20170707.jpg':42.5587611}\n",
"\n",
"latlongWTs2=[]\n",
"for ll in range(len(detectedWTs)):\n",
" a5=chipsize*(detectedWTs[ll][1]-1+centerBBs[ll][0])\n",
" b5=chipsize*(detectedWTs[ll][2]-1+centerBBs[ll][1])\n",
"\n",
" p_a5=a5/width\n",
" p_b5=b5/height\n",
"\n",
" x_le=x1[detectedWTs[ll][0]]+p_b5*(x4[detectedWTs[ll][0]]-x1[detectedWTs[ll][0]])\n",
" x_re=x2[detectedWTs[ll][0]]+p_b5*(x3[detectedWTs[ll][0]]-x2[detectedWTs[ll][0]])\n",
" x_WT=x_le+p_a5*(x_re-x_le)\n",
"\n",
" y_te=y1[detectedWTs[ll][0]]+p_a5*(y2[detectedWTs[ll][0]]-y1[detectedWTs[ll][0]])\n",
" y_be=y4[detectedWTs[ll][0]]+p_a5*(y3[detectedWTs[ll][0]]-y4[detectedWTs[ll][0]])\n",
" y_WT=y_te+p_b5*(y_be-y_te)\n",
" latlongWTs2.append([y_WT,x_WT])\n",
"\n",
"print(latlongWTs2)\n",
"print(len(latlongWTs2))"
],
"execution_count": 0,
"outputs": []
Expand Down

0 comments on commit 24518dd

Please sign in to comment.