Convert txt to pkl #24

Activity

1. quangnmkma commented on Dec 9, 2022

   

   quangnmkma

   on Dec 9, 2022

   Last edited by quangnmkma

   More actions

   I have the same the question :((
2. luigy-mach commented on Dec 12, 2022

   

   luigy-mach

   on Dec 12, 2022

   More actions

   In file quick-start.ipynb comment the next lines as follows:

   # bboxes = pkl.load(open("messi_ann.pkl", "rb"))
   # #inspect the bounding boxes
   # print(bboxes)
   

   And, add the next lines:

   in_coor  = ['class','x1','y1','x2','y2'] # how in your file was sorted.
   out_coor = ['x1','y1','x2','y2','class'] # how it was requested by algorithm.
   filetxt  = 'yourfile.txt' # path of your file txt 
   with open(filetxt) as f:
       contents = f.readlines()
       
   data = list()
   for i in contents:
       data.append([float(j) for j in i.split(',')])
   
   data_arr  = np.asarray(data)
   in_order  = {j:i for i,j in enumerate(in_coor)}
   sort_in   = [in_order[k] for k in in_coor]
   sort_out  = [in_order[k] for k in out_coor]
   
   bboxes = data_arr[:,sort_out]
   #inspect the bounding boxes
   
   print(bboxes)