added classes.pbxtx, created csv:s from annotations

classes.pbtxt

+item {
+    id: 1
+    name: ‘ladybug’
+    id: 2
+    name: ‘ant’
+}
\ No newline at end of file

data/test_labels.csv

+filename,width,height,class,xmin,ymin,xmax,ymax
+62.jpg,641,1024,ant,101,578,255,687
+52.jpg,1024,604,ant,0,888,76,524
+34.jpg,1024,680,ant,0,639,154,568
+34.jpg,1024,680,ant,727,1023,55,343
+117.jpg,1024,772,ladybug,245,795,166,643
+91.jpg,1024,683,ladybug,206,811,96,549
+63.jpg,682,1024,ant,58,550,68,879
+35.jpg,1024,683,ant,58,284,424,536
+35.jpg,1024,683,ant,85,431,511,644
+35.jpg,1024,683,ant,306,629,174,398
+35.jpg,1024,683,ant,645,965,142,288
+17.jpg,1024,682,ant,392,683,222,485
+15.jpg,1024,898,ant,383,570,235,482
+121.jpg,1024,620,ladybug,461,609,316,518
+82.jpg,1024,578,ladybug,602,819,116,412
+116.jpg,1024,683,ladybug,302,359,373,423
+95.jpg,1024,683,ladybug,704,782,522,602
+95.jpg,1024,683,ladybug,771,826,545,606
+95.jpg,1024,683,ladybug,848,896,288,322
+141.jpg,1024,685,ladybug,669,803,302,398
+61.jpg,1024,683,ant,258,802,195,469
+143.jpg,1024,768,ladybug,460,727,309,680
+66.jpg,1024,680,ant,19,229,370,547
+66.jpg,1024,680,ant,30,231,117,268
+66.jpg,1024,680,ant,206,312,348,508
+55.jpg,1024,768,ant,124,229,432,527
+55.jpg,1024,768,ant,492,620,539,668
+55.jpg,1024,768,ant,679,803,560,637
+78.jpg,1024,633,ant,243,611,111,458
+7.jpg,1024,683,ant,347,521,347,449
+7.jpg,1024,683,ant,718,880,228,343
+86.jpg,1024,596,ant,305,810,125,451
+156.jpg,682,1024,ladybug,399,519,574,701
+151.jpg,728,1024,ladybug,147,478,371,672
+76.jpg,1024,682,ant,192,900,13,681
+164.jpg,1024,683,ladybug,438,501,307,381
+125.jpg,1024,577,ladybug,423,483,227,299
+125.jpg,1024,577,ladybug,463,529,250,324
+98.jpg,1024,683,ladybug,413,638,218,412
+134.jpg,1024,768,ladybug,456,512,355,410
+45.jpg,777,1024,ant,6,753,7,1000
+159.jpg,1024,821,ladybug,209,658,66,745
+153.jpg,1024,768,ladybug,501,698,259,404
+122.jpg,1024,682,ladybug,457,564,293,387
+154.jpg,1024,683,ladybug,255,421,245,378
+42.jpg,1024,769,ant,101,459,386,726
+42.jpg,1024,769,ant,642,1016,21,414
+135.jpg,1024,768,ladybug,238,905,97,602
+165.jpg,1024,768,ladybug,319,663,237,542
+84.jpg,1024,683,ladybug,405,515,237,328
+145.jpg,1024,768,ladybug,270,664,202,473
+3.jpg,1024,798,ant,213,351,494,577
+139.jpg,1024,768,ladybug,354,633,314,599
+127.jpg,1024,622,ladybug,184,346,250,382
+109.jpg,915,1024,ladybug,98,473,246,550
+90.jpg,1024,683,ladybug,322,723,141,449
+129.jpg,768,1024,ladybug,16,81,399,448
+129.jpg,768,1024,ladybug,19,47,17,38
+129.jpg,768,1024,ladybug,96,117,12,36
+129.jpg,768,1024,ladybug,102,144,163,198
+129.jpg,768,1024,ladybug,297,346,162,199
+129.jpg,768,1024,ladybug,307,382,597,647
+129.jpg,768,1024,ladybug,355,422,548,616
+129.jpg,768,1024,ladybug,557,625,359,408
+72.jpg,1024,849,ant,336,901,26,590
+142.jpg,682,1024,ladybug,378,534,664,828
+148.jpg,1024,768,ladybug,446,638,303,410
+71.jpg,1024,768,ant,109,586,256,415
+59.jpg,1024,829,ant,58,929,197,673
+1.jpg,1024,719,ant,159,303,313,361
+49.jpg,1024,768,ant,92,792,173,616
+158.jpg,1024,1024,ladybug,50,987,145,853
+67.jpg,1024,1024,ant,133,817,0,888
+65.jpg,1024,683,ant,108,817,7,553
+94.jpg,1024,686,ladybug,434,513,263,371
+4.jpg,1024,682,ant,379,544,173,369
+4.jpg,1024,682,ant,476,705,386,620
+47.jpg,1024,768,ant,340,478,375,490
+47.jpg,1024,768,ant,455,620,320,425
+161.jpg,683,1024,ladybug,151,475,192,771
+64.jpg,1024,768,ant,180,215,514,534
+64.jpg,1024,768,ant,182,215,583,611
+64.jpg,1024,768,ant,321,345,421,441
+64.jpg,1024,768,ant,323,347,442,468
+64.jpg,1024,768,ant,349,357,433,446
+64.jpg,1024,768,ant,364,389,427,439
+64.jpg,1024,768,ant,371,398,534,546
+64.jpg,1024,768,ant,388,423,557,575
+64.jpg,1024,768,ant,428,451,608,640
+64.jpg,1024,768,ant,436,467,649,669
+64.jpg,1024,768,ant,479,508,678,703
+64.jpg,1024,768,ant,508,540,346,372
+64.jpg,1024,768,ant,562,585,562,594
+64.jpg,1024,768,ant,638,657,448,479
+64.jpg,1024,768,ant,660,681,172,206
+64.jpg,1024,768,ant,712,746,255,277
+64.jpg,1024,768,ant,744,766,576,609
+64.jpg,1024,768,ant,938,967,256,272
+108.jpg,1024,682,ladybug,480,665,379,516
+140.jpg,896,1024,ladybug,343,504,550,642
+81.jpg,1024,393,ant,170,970,49,312
+114.jpg,1024,768,ladybug,393,611,286,426
+30.jpg,1024,612,ant,55,707,83,497
+30.jpg,1024,612,ant,553,1016,182,569
+23.jpg,1024,647,ant,33,346,188,423
+23.jpg,1024,647,ant,74,382,366,596
+23.jpg,1024,647,ant,305,650,235,413
+18.jpg,1024,683,ant,424,545,290,447
+31.jpg,1024,768,ant,440,1004,187,626
+2.jpg,1024,646,ant,108,784,125,332
+147.jpg,1024,809,ladybug,186,469,252,533
+22.jpg,1024,653,ant,3,933,53,467
+37.jpg,986,1024,ant,84,797,93,933
+10.jpg,1024,682,ant,144,895,261,625
+40.jpg,1024,849,ant,147,426,400,566
+54.jpg,1024,1024,ant,84,963,25,792
+131.jpg,1024,680,ladybug,432,551,263,392
+128.jpg,1024,576,ladybug,195,410,137,399
+60.jpg,1024,734,ant,258,954,148,676
+137.jpg,1024,971,ladybug,138,438,292,563
+44.jpg,1024,768,ant,313,691,24,372
+44.jpg,1024,768,ant,673,931,103,410
+19.jpg,685,1024,ant,76,658,120,881
+8.jpg,1024,683,ant,69,664,347,680
+157.jpg,1024,681,ladybug,666,723,305,364
+24.jpg,1024,679,ant,74,213,123,249
+24.jpg,1024,679,ant,167,351,236,383
+24.jpg,1024,679,ant,491,726,198,345
+24.jpg,1024,679,ant,766,1023,197,323
+9.jpg,1024,768,ant,435,592,353,531
+9.jpg,1024,768,ant,12,204,618,767
+9.jpg,1024,768,ant,326,421,108,234
+102.jpg,1024,683,ladybug,448,564,299,403
+124.jpg,683,1024,ladybug,279,527,533,725
+43.jpg,1024,686,ant,402,958,237,543
+41.jpg,1024,683,ant,224,864,131,503
+39.jpg,1024,532,ant,510,584,331,416
+39.jpg,1024,532,ant,97,191,311,375
+39.jpg,1024,532,ant,156,268,345,405
+39.jpg,1024,532,ant,248,320,250,302
+39.jpg,1024,532,ant,387,506,250,324
+39.jpg,1024,532,ant,403,517,400,492
+39.jpg,1024,532,ant,529,618,239,318
+39.jpg,1024,532,ant,826,920,249,281
+48.jpg,1024,727,ant,174,339,479,611
+48.jpg,1024,727,ant,412,671,252,427
+48.jpg,1024,727,ant,659,800,6,158
+48.jpg,1024,727,ant,748,960,60,236
+48.jpg,1024,727,ant,837,1000,321,500
+167.jpg,1024,768,ladybug,424,505,332,454
+87.jpg,1024,896,ladybug,340,653,192,584
+97.jpg,1024,1024,ladybug,361,583,500,656
+144.jpg,1024,596,ladybug,566,692,288,366
+103.jpg,685,1024,ladybug,263,368,465,567
+74.jpg,1024,768,ant,1,1022,140,666
+12.jpg,1024,819,ant,78,508,91,670
+12.jpg,1024,819,ant,542,905,164,783
+118.jpg,1024,845,ladybug,323,689,221,589
+33.jpg,1024,767,ant,140,549,263,645
+111.jpg,1024,768,ladybug,472,601,292,436
+28.jpg,1024,768,ant,143,823,12,761
+80.jpg,1024,768,ant,76,938,147,711
+56.jpg,682,1024,ant,208,498,261,708
+16.jpg,1024,778,ant,421,612,268,489
+36.jpg,1024,768,ant,228,350,452,577
+36.jpg,1024,768,ant,311,413,76,177
+36.jpg,1024,768,ant,336,725,110,696
+36.jpg,1024,768,ant,727,848,222,363
+73.jpg,1024,768,ant,223,904,169,720
+5.jpg,1024,804,ant,248,616,283,671
+5.jpg,1024,804,ant,263,646,66,326
+101.jpg,1024,841,ladybug,457,598,226,347
+70.jpg,1024,861,ant,364,673,203,570
+150.jpg,1024,768,ladybug,548,734,275,405
+120.jpg,1024,703,ladybug,231,770,84,670
+0.jpg,1024,768,ant,17,334,238,568
+0.jpg,1024,768,ant,342,723,116,540
+119.jpg,1024,731,ladybug,65,468,168,541
+169.jpg,1024,887,ladybug,362,666,90,350
+20.jpg,1024,732,ant,290,705,94,725
+88.jpg,1024,700,ladybug,538,568,248,273
+155.jpg,1024,768,ladybug,265,771,441,626
+110.jpg,1024,768,ladybug,120,440,238,508
+6.jpg,1024,683,ant,359,892,112,374
+83.jpg,1024,768,ant,165,844,129,696
+152.jpg,1024,898,ladybug,98,878,78,752
+160.jpg,1024,576,ladybug,159,503,119,490
+50.jpg,1024,768,ant,0,508,154,558
+50.jpg,1024,768,ant,487,650,0,256
+50.jpg,1024,768,ant,524,797,229,474
+50.jpg,1024,768,ant,547,824,544,767
+38.jpg,1024,683,ant,16,54,460,472
+38.jpg,1024,683,ant,112,138,526,550
+38.jpg,1024,683,ant,138,168,474,509
+38.jpg,1024,683,ant,139,176,345,379
+38.jpg,1024,683,ant,162,198,436,463
+38.jpg,1024,683,ant,186,224,359,377
+38.jpg,1024,683,ant,203,245,472,491
+38.jpg,1024,683,ant,222,256,377,396
+38.jpg,1024,683,ant,249,289,585,610
+38.jpg,1024,683,ant,249,275,461,481
+38.jpg,1024,683,ant,266,305,507,521
+38.jpg,1024,683,ant,318,354,510,525
+38.jpg,1024,683,ant,424,455,420,454
+38.jpg,1024,683,ant,648,686,520,559
+38.jpg,1024,683,ant,706,745,557,593
+38.jpg,1024,683,ant,876,918,478,512
+38.jpg,1024,683,ant,952,988,498,534
+29.jpg,768,1024,ant,88,721,124,916
+162.jpg,1024,768,ladybug,407,553,313,457
+77.jpg,1024,820,ant,527,816,448,597
+112.jpg,1024,645,ladybug,712,896,299,417
+123.jpg,1024,710,ladybug,432,715,203,475
+27.jpg,1024,683,ant,150,321,23,175
+27.jpg,1024,683,ant,476,663,276,459
+149.jpg,1024,683,ladybug,391,526,235,408
+104.jpg,768,1024,ladybug,101,271,153,363
+138.jpg,1024,768,ladybug,486,564,332,398
+89.jpg,1024,767,ladybug,446,626,224,372
+166.jpg,1024,821,ladybug,618,755,291,472
+113.jpg,957,768,ladybug,440,595,337,490
+113.jpg,957,768,ladybug,310,404,137,251
+133.jpg,1024,886,ladybug,137,410,413,800
+163.jpg,1024,682,ladybug,478,552,323,402
+146.jpg,1024,683,ladybug,383,666,204,451
+136.jpg,1024,680,ladybug,244,851,151,570
+96.jpg,1024,683,ladybug,559,821,345,581
+115.jpg,1024,834,ladybug,279,757,263,562
+75.jpg,1024,768,ant,60,620,115,462
+11.jpg,1024,744,ant,478,670,129,222
+11.jpg,1024,744,ant,266,393,167,272
+51.jpg,1024,686,ant,59,936,140,612
+14.jpg,1024,681,ant,380,698,260,476
+68.jpg,1024,682,ant,373,971,235,615
+46.jpg,1024,684,ant,64,891,14,569
+107.jpg,1024,795,ladybug,430,755,214,538
+99.jpg,1024,768,ladybug,401,611,309,500
+79.jpg,1024,768,ant,16,138,108,216
+92.jpg,1024,900,ladybug,220,460,366,562
+92.jpg,1024,900,ladybug,502,714,263,497
+58.jpg,1024,847,ant,158,841,112,817
+85.jpg,1024,768,ant,358,624,261,590
+13.jpg,1024,683,ant,54,924,187,576
+32.jpg,1024,824,ant,250,792,22,799
+126.jpg,768,1024,ladybug,113,556,151,662
+57.jpg,1024,682,ant,74,903,160,677
+130.jpg,1024,767,ladybug,431,558,422,523
+25.jpg,1024,768,ant,8,388,25,170
+25.jpg,1024,768,ant,121,264,277,480
+25.jpg,1024,768,ant,277,411,533,755
+25.jpg,1024,768,ant,388,640,255,499
+69.jpg,1024,768,ant,67,222,391,577
+69.jpg,1024,768,ant,486,662,341,555
+69.jpg,1024,768,ant,576,719,223,328
+69.jpg,1024,768,ant,614,693,375,502
+69.jpg,1024,768,ant,704,848,348,554
+100.jpg,1024,640,ladybug,269,570,264,489
+168.jpg,1024,921,ladybug,320,862,184,684
+132.jpg,1024,786,ladybug,447,603,334,559
+93.jpg,1024,667,ladybug,269,863,142,577
+21.jpg,1024,682,ant,128,899,46,648
+105.jpg,1020,1024,ladybug,271,393,468,611
+106.jpg,1024,688,ladybug,674,795,246,389
+53.jpg,1024,609,ant,569,962,253,495
+26.jpg,1024,768,ant,393,663,339,492
+26.jpg,1024,768,ant,109,389,436,597

xmltocsv.py

+import os
+import glob
+import pandas as pd
+import xml.etree.ElementTree as ET
+
+
+def xml_to_csv(path):
+    xml_list = []
+    for xml_file in glob.glob(path + '/*.xml'):
+        tree = ET.parse(xml_file)
+        root = tree.getroot()
+        for member in root.findall('object'):
+            value = (root.find('filename').text,
+                     int(root.find('size')[0].text),
+                     int(root.find('size')[1].text),
+                     member[0].text,
+                     int(member[4][0].text),
+                     int(member[4][1].text),
+                     int(member[4][2].text),
+                     int(member[4][3].text)
+                     )
+            xml_list.append(value)
+    column_name = ['filename', 'width', 'height', 'class', 'xmin', 'ymin', 'xmax', 'ymax']
+    xml_df = pd.DataFrame(xml_list, columns=column_name)
+    return xml_df
+
+
+def main():
+	'''
+    for directory in ['train','testing']:
+        image_path = os.path.join(os.getcwd(), 'images/{}'.format(directory).format(directory))
+        xml_df = xml_to_csv(image_path)
+        xml_df.to_csv('data/{}_labels.csv'.format(directory), index=None)
+        print('Successfully converted xml to csv.')
+	''' 
+	image_path = os.path.join(os.getcwd(), 'data/train/Annotations')
+	xml_df = xml_to_csv(image_path)
+	xml_df.to_csv('data/train_labels.csv', index=None)
+
+	image_path = os.path.join(os.getcwd(), 'data/test/Annotations')
+	xml_df = xml_to_csv(image_path)
+	xml_df.to_csv('data/test_labels.csv',index=None)
+
+main()
\ No newline at end of file