Commit Graph

2238 Commits

Author SHA1 Message Date
AlexeyAB d18e22ae1b fix new_coords for CPU 2020-11-20 21:14:31 +03:00
AlexeyAB d82cf7e684 max_delta=20 for yolov4x-mish.cfg 2020-11-20 18:45:30 +03:00
AlexeyAB 9db8609e7a Added cspx-p7-mish.cfg 2020-11-20 04:25:41 +03:00
AlexeyAB 5c046bea96 minor fix 2020-11-18 04:06:28 +03:00
AlexeyAB 17cf4bfbfc added yolov4x-mish.cfg 2020-11-16 20:36:02 +03:00
AlexeyAB 8c9c517189 [yolo] new_coords=1 2020-11-16 16:29:06 +03:00
AlexeyAB c7e3e2ee9e Fix nms for valid, coco 2014 -> 2017 2020-11-16 15:17:18 +03:00
Alexey b51cf902ee
Merge pull request #6943 from MarvinKlemp/master
Added a hint for installation with CUDA version >= 11
2020-11-15 23:21:12 +03:00
Alexey 46a168bfae
Merge pull request #6944 from Sialitech/classifier
Restore classifier function to python API
2020-11-15 23:21:01 +03:00
MarvinKlemp fe8e9edd29
Added Ampere Gencode 2020-11-12 10:12:50 +01:00
MarvinKlemp 3d33bbfecc
generalized statement 2020-11-12 10:07:54 +01:00
JaledMC 682210bc72
Free image memory 2020-11-09 00:36:36 +01:00
JaledMC 236205879d
Add image_classification function to darknet_images
Enable darknet classifier, without bbox detection
2020-11-09 00:31:42 +01:00
JaledMC 3817006ef9 Sync with original repo 2020-11-09 00:24:37 +01:00
MarvinKlemp 5ebcf63fce
Added a hint for installation with CUDA version >= 11 2020-11-08 16:30:21 +01:00
Alexey 14b196d4f2
Update README.md 2020-11-04 22:01:46 +03:00
AlexeyAB f04f418d27 cspx-p7-mish_hp.cfg experimental 2020-11-03 20:28:52 +03:00
AlexeyAB 8297b1be31 Added some hyper-params to the yolov4-tiny_contrastive.cfg 2020-10-20 05:37:49 +03:00
AlexeyAB 88936e06a9 Added some hyper-params to the yolov4-tiny_contrastive.cfg 2020-10-20 05:34:37 +03:00
AlexeyAB 7a28aade28 yolov4-tiny_contrastive.cfg 2020-10-20 02:49:39 +03:00
Alexey d65909fbea
Update README.md 2020-10-11 15:54:55 +03:00
Alexey 91b5dd2da7
Update README.md 2020-10-07 01:14:00 +03:00
Alexey f8afd1b016
Merge pull request #6719 from cenit/dev/cenit/llvm12
fix compatibility with newer clang
2020-10-05 06:46:27 +03:00
Alexey 9db0ed9662
Merge pull request #6642 from zk2ly/zk2ly-patch-1
Update darknet_images.py
2020-09-25 07:10:57 +03:00
Alexey 622b175860
Merge pull request #6668 from Shaing/master
Fix: Python example API that caused Memory leak.
2020-09-25 07:03:16 +03:00
Stefano Sinigardi 860d5291d4 fix compatibility with newer clang 2020-09-23 11:03:47 +02:00
AlexeyAB fdb1841eb1 Ignore behavior of [history] layer for training on non-sequential images 2020-09-21 06:11:20 +03:00
AlexeyAB 8993a24747 Added: [net] contrastive_color=0 and [contrastive] contrastive_neg_max 2020-09-21 05:48:15 +03:00
AllenH 5aa9fddce6 fix: python API that caused Memory leak 2020-09-15 15:47:05 +08:00
zk2ly 05a88b2614
Update darknet_images.py 2020-09-10 15:28:21 +08:00
AlexeyAB eb0272f27a Fixed cuDNN 8.x usage 2020-09-06 04:54:20 +03:00
AlexeyAB ecad770071 Added [yolo] l.cls_normalizer 2020-08-30 02:34:29 +03:00
AlexeyAB acd8f31969 Added [yolo] obj_normalizer=1.0 delta_normalizer=1.0 2020-08-30 01:55:49 +03:00
Alexey ae27c1652b
Merge pull request #6564 from bouncyphoton/master
Improve model loading performance for detect
2020-08-29 17:52:06 +03:00
JaledMC d7deb35f27
Merge pull request #7 from AlexeyAB/master
Update our fork with alexeyab advances
2020-08-25 08:50:50 +02:00
sandy 519fc8b11b Remove expensive unnecessary memset 2020-08-18 16:57:02 -07:00
sandy 6a2d50f243 Precompute part of expensive multiplication 2020-08-18 16:56:45 -07:00
AlexeyAB 05dee78fa3 Added [convolutional] coordconv=1 for GPU-only 2020-08-17 02:13:46 +03:00
AlexeyAB 0d8718a434 Added layer: [history] history_size=4 2020-08-16 17:55:10 +03:00
Alexey 38a164bcb9
Update README.md 2020-08-11 21:01:40 +03:00
Alexey a2ccbd47ff
Update Readme.md 2020-08-10 16:15:25 +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
Alexey af08480df4
Merge pull request #6441 from dontcare/master
Update darknet.py to run outside the darknet folder
2020-08-09 15:31:54 +03:00
Alexey f4c373e9fe
Merge pull request #5717 from cenit/dev/cenit/cuda-cmake
small fixes and improvements
2020-08-09 15:31:05 +03:00
Stefano Sinigardi 86b1bf7497 restore missing infos in README.md after merging with master 2020-08-08 16:25:53 +02:00
Stefano Sinigardi 8cddef3466 upgrade vcpkg commit ref 2020-08-08 15:35:57 +02:00
Stefano Sinigardi c00dbbb9fa bump minimum compute capability to 5.0 2020-08-08 15:35:46 +02:00
Stefano Sinigardi 78293335df unify lib naming with makefile 2020-08-08 13:15:03 +02:00
Stefano Sinigardi 20968ef98b Merge branch 'dev/alexey/master' into dev/cenit/cuda-cmake 2020-08-08 13:10:27 +02: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