utils_os.py

Go to the documentation of this file.

import sys
import argparse
import os
import random
import operator
import json
import numpy
 
 
class MyEncoder(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, numpy.integer):
            return int(obj)
        elif isinstance(obj, numpy.floating):
            return float(obj)
        elif isinstance(obj, numpy.ndarray):
            return obj.tolist()
        else:
            return super(MyEncoder, self).default(obj)
 
 
def save_keep_index(sorted_keep_index, crop_image_path, test_image_list, save_path):
    save_file = os.path.join(save_path,'keep_index.json')
    data = {}
    data['image_path'] = []
    data['s_sim'] = []
    for index in sorted_keep_index:
        image_path = os.path.join(crop_image_path, test_image_list[index[0]])
        data['image_path'].append(image_path)
        data['s_sim'].append(index[1])
        print(f'image_path = {image_path}')
        print(f's_sim = {index[1]}')
 
    with open(save_file, 'w') as outfile:
        json.dump(data, outfile,cls=MyEncoder)