Outils pour utilisateurs

Outils du site


darknet_letters_unload_gpu_ram_in_python_script

Ceci est une ancienne révision du document !


Darknet Letters unload GPU RAM in python script

Modifications des fichiers

./include/darknet.h

void show_total_time();
 
// ajout alexandre pour vider la memoire
LIB_API void free_network(network net);
 
#ifdef __cplusplus
}
#endif  // __cplusplus
#endif  // DARKNET_API

./src/network.c

 

./src/network.h

 

./darknet.py

predict_image.restype = POINTER(c_float)
 
# alexandre lien entre lib api de darknet.c et play_letters
free_network = lib.free_network
free_network.argtypes = [c_void_p]
 
def array_to_image(arr):

Dans votre script personnel qui appelle darknet.py

Exemple: voir le fichier complet play_letters.py sur GitHub

class YOLO:
    ...
    ...
    def set_darknet(self):
        ...
        ...
        # Définition du network
        self.netMain = darknet.load_net_custom(configPath.encode("ascii"),
                                                weightPath.encode("ascii"),
                                                0,
                                                1)

En fin de boucle opencv:

            # Echap pour finir le script python
            if cv2.waitKey(1) == 27:
                os._exit(0)
 
        cv2.destroyAllWindows()
 
        # Libération de la mémoire GPU
        self.free_network(self.netMain)
darknet_letters_unload_gpu_ram_in_python_script.1575044964.txt.gz · Dernière modification : 2019/11/29 16:29 de serge