Преобразуйте данные в новый формат для обнаружения объектов - PullRequest
0 голосов
/ 11 февраля 2020

У меня есть набор данных в этом формате в фрейме данных

0--Parade/0_Parade_marchingband_1_849.jpg
2
449 330 122 149 0 0 0 0 0 0 
0--Parade/0_Parade_Parade_0_904.jpg
1
361 98 263 339 0 0 0 0 0 0 
0--Parade/0_Parade_marchingband_1_799.jpg
45
78 221 7 8   0 0 0 0 0 
78 238 14 17 2 0 0 0 0 0  
3 232 11 15 2 0 0 0 2 0 
20 215 12 16 2 0 0 0 2 0 
0--Parade/0_Parade_marchingband_1_117.jpg
23
69 359 50 36 1 0 0 0 0 1 
227 382 56 43 1 0 1 0 0 1 
296 305 44 26 1 0 0 0 0 1 
353 280 40 36 2 0 0 0 2 1 
885 377 63 41 1 0 0 0 0 1 
819 391 34 43 2 0 0 0 1 0 
727 342 37 31 2 0 0 0 0 1 
598 246 33 29 2 0 0 0 0 1 
740 308 45 33 1 0 0 0 2 1 
0--Parade/0_Parade_marchingband_1_778.jpg
35
27 226 33 36 1 0 0 0 2 0 
63 95 16 19 2 0 0 0 0 0 
64 63 17 18 2 0 0 0 0 0 
88 13 16 15 2 0 0 0 1 0 
231 1 13 13 2 0 0 0 1 0 
263 122 14 20 2 0 0 0 0 0 
367 68 15 23 2 0 0 0 0 0 
198 98 15 18 2 0 0 0 0 0 
293 161 52 59 1 0 0 0 1 0 
412 36 14 20 2 0 0 0 1 0 

Может кто-нибудь сказать мне, как поместить их в фрейм данных, где 1-й столбец содержит все пути .jpg Следующий столбец содержит все координаты, но все координата должна соответствовать этому .jpg пути

например.

             Column1                          coulmn2      column3
0--Parade/0_Parade_marchingband_1_849.jpg     |  2  |  449 330 122 149 0 0 0 0 0 0 

0--Parade/0_Parade_Parade_0_904.jpg           |  1  |  361 98 263 339 0 0 0 0 0 0 

0--Parade/0_Parade_marchingband_1_799.jpg     |  45 |   78 221 7 8   0 0 0 0 0 
                                              |     |   78 238 14 17 2 0 0 0 0 0  
                                              |     |   3 232 11 15 2 0 0 0 2 0 
                                              |     |   20 215 12 16 2 0 0 0 2 0 

Я пробовал это

count1=0
count2=0
dict1 = {}
dict2 = {}
dict3 = {}
for i in data[0]:
    if (i.find('.jpg') == -1): 
        dict1[count1] = i
        count1+=1

    else: 
        dict2[count2] = i
        count2+=1
...