diff --git a/read_dataset.py b/read_dataset.py new file mode 100644 index 0000000000000000000000000000000000000000..a2057cb224cd33a321a0799afc194c27fd16e3e9 --- /dev/null +++ b/read_dataset.py @@ -0,0 +1,89 @@ +import pandas as pd +import json + +#df = pd.read_json('/media/gduret/DATA/dataset/s2rg/Fruits_all_medium/data/1/meta.json') + +#print(df.to_string()) + + +#df = pd.DataFrame([1,2,3]) + +#df.to_json('example.json') + + +#Category = "kiwi1" +#Category = "cherry1" +#Category = "crate" +#Category = "banana1" +#Category = "apricot" +#Category = "orange2" +#Category = "pear2" +#Category = "floor" +#Category = "peach1" +#Category = "lemon2" +#Category = "apple2" +#Category = "strawberry1" + +#Category = 'banana1' +Category = 'pear2' + +# Opening JSON file +f = open('/media/gduret/DATA/dataset/s2rg/Fruits_all_medium/data/6/meta.json') + +# returns JSON object as +# a dictionary +data = json.load(f) + +# Iterating through the json +# list + +id_category = 0 + +for i in data['categories']: + print(i) + if i['label'] == Category : + print("Hello fruits") + id_category = i['id'] + print(id_category) + +id_instances = [] + +for i in data['objects']: + print(i) + if i['category_id'] == id_category : + print("Hello fruits instance") + id_instances.append(i['id']) + print(i['id']) + +print(id_instances) + +# Closing file +f.close() + +f2 = open('/media/gduret/DATA/dataset/s2rg/Fruits_all_medium/data/6/grabber_1/ground_truth_rgb/occlusion/4_158000000.json') + +data2 = json.load(f2) + +id_instances_good = [] + +for i in data2: + print('i : ',i) + print(i['id']) + print(id_instances) + + if i['id'] in id_instances : + print("Hello banana instance occ") + if i['occlusion_value'] > 0.5 : + id_instances_good.append(i['id']) + print(i['id']) + print(i['occlusion_value']) + +print(id_instances_good) + + + + +# Closing file +f2.close() + +