darknet_letters_unload_gpu_ram_in_python_script
Ceci est une ancienne révision du document !
Darknet Letters unload GPU RAM in python script
- Bug: Libération de la RAM GPU à la fin d'une détection en gardant le script python actif.
- unload network in python script https://github.com/AlexeyAB/darknet/issues/3467
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