Commit Graph

40 Commits

Author SHA1 Message Date
Sarim Mehdi 9ff8653d99
Made inference faster (this is especially useful when using Yolo9000) (#8009)
* Update network.c

custom_get_region_detections function now keeps track of class index with the highest probability.

* Update darknet.h

Added best_class_idx to detection struct

* Update darknet.py

added python code for faster negative removal and also faster non-max suppression
2021-08-25 18:10:21 +03:00
Stefano Sinigardi 9c26b291fa
fix python main module to be compiled with recent build scripts (#7876)
* fix python main module to be compiled with recent build scripts

* fixes for posix systems
2021-07-09 14:50:38 +03:00
Alexey 9dc897d2c7
Merge pull request #6440 from guitarmind/master
Add free_network_ptr Support for Darknet in Python
2020-08-09 15:32:12 +03:00
Dontcare 9ecc4f7d6b
Update darknet.py
# Example code

import os
import sys

os.environ['DARKNET_PATH'] = '/usr/local/darknet'
sys.path.append(os.environ['DARKNET_PATH'])

import darknet


def main():
    network, class_names, class_colors = darknet.load_network(
        '../nscan-train/data/cfg/yolov4-tiny-3l-numbers-12.cfg',
        '../nscan-train/data/numbers.data',
        './data/weights/numbers/yolov4-tiny-3l-numbers-12_last.weights',
        batch_size=1
    )


if __name__ == '__main__':
    main()
2020-08-08 12:46:20 +03:00
Mark Peng 59bb012876 Merge remote-tracking branch 'upstream/master' 2020-08-08 12:59:34 +08:00
Mark Peng 725d9092e4 Added free_network_ptr() support for darknet.py. 2020-08-08 12:57:52 +08:00
JaledMC d0b6bae731
Update darknet.py 2020-07-29 17:35:39 +02:00
JaledMC 1bb2d70bf9
Update darknet.py 2020-07-29 17:34:03 +02:00
JaledMC 1360070bf5
Update darknet.py 2020-07-29 17:17:40 +02:00
AlexeyAB ccb392ddf2 Show track_id for models with contrastive learning 2020-07-07 19:30:49 +03:00
JaledMC 52c7314340 python wrapper cleaning and new features
a darknet module and both scripts for video and image inference, with the features of C code
2020-05-01 18:30:53 +02:00
AlexeyAB 6d38218a04 Minor fix 2020-04-28 19:20:22 +03:00
AlexeyAB b573eab63f Minor fix 2020-04-22 20:49:27 +03:00
Alexey 88f28f7fcc
Merge pull request #4099 from enesozi/master
Batch inference added
2020-04-22 14:46:38 +03:00
AlexeyAB 48914023da Added init_cpu() function to init CPU just to speedup the 1st detection (if GPU=0) 2019-11-25 18:10:39 +03:00
AlexeyAB 3abbd85808 fixed consistency darknet and python code 2019-11-23 20:00:35 +03:00
AlexeyAB d628e8eab7 Fixed darknet.py for Uncertainty (gaussian_yolo_layer) 2019-11-05 16:57:03 +03:00
enes b392621e2e batch inference refactoring 2019-10-19 16:18:44 +03:00
enes 1108eafa95 Batch inference added 2019-10-16 10:00:57 +03:00
AlexeyAB 54e2d0b0e8 Added ability to use letter_box resizing in darknet.py sample 2019-07-01 19:07:13 +03:00
AlexeyAB 9bb7455a0e Minor fix 2019-02-07 14:47:43 +03:00
AlexeyAB c50b0e0c8a Minor Python and C API improvement 2019-02-06 14:38:12 +03:00
John Aughey 7e9416aa80 Making a pointer version of network_predict for python.
The python binding requires the network struct to be passed
as a pointer to it rather than a struct copy.
2019-02-05 11:35:45 -06:00
AlexeyAB 3ff5084590 Added include/darknet.h 2019-01-06 23:51:38 +03:00
Vincent Liao f81b9209ff fix wrong path 2018-10-24 12:47:19 +07:00
AlexeyAB 026a679ded Minor fix: darknet.py for OpenCV 2018-06-21 19:23:30 +03:00
AlexeyAB 56bff1dc8c Fixed darknet.py when OpenCV is used (added BGR to RGB conversion) 2018-06-20 01:06:39 +03:00
AlexeyAB 1e512b2308 Letter box is disabled in the darknet.py by defualt 2018-06-04 22:11:05 +03:00
Vinjn Zhang dbcdac88a5
darknet.py - loading dll files by absolute name 2018-05-20 14:20:55 +08:00
Vinjn Zhang efdeb01a3a
darknet.py - remove unnecessary check of CUDA_HOME 2018-05-20 13:57:21 +08:00
AlexeyAB 24f563ce71 Example of usage cv2.imread() in the darknet.py 2018-05-17 23:26:55 +03:00
AlexeyAB b34c913d2e darknet.py change libdarknet.so to darknet.so 2018-05-16 15:16:41 +03:00
AlexeyAB 89c11f83ed Fixed darknet.py - uses batch=1 by default 2018-05-16 15:05:46 +03:00
AlexeyAB 573d7e8081 Added def array_to_image(arr): to the darknet.py
Usage:
    import scipy.misc
    sci_image = scipy.misc.imread(image)
    im, arr = array_to_image(sci_image)
2018-05-16 03:03:23 +03:00
Philip Kahn 51fbf01277
Address GPU message in root dir 2018-05-04 15:07:45 -07:00
Philip Kahn 3a2a8653a6
Fix detection paths 2018-05-04 14:59:29 -07:00
Philip Kahn 9c400dc5b8
Remove debugging line 2018-05-04 14:46:33 -07:00
Philip Kahn 3459c47bfa
Also replace root version to Python3 2018-05-04 14:00:11 -07:00
AlexeyAB 408bde78ff Fixed darknet.py for Linux 2018-04-13 17:32:10 +03:00
AlexeyAB fc496d52bf Yolo can be used from Python 2.x using /darknet.py on Linux or darknet/x64/darknet.py on Windows 2018-04-12 22:32:57 +03:00