The library reproduces the reproduction of neural networks on the Darknet framework in ClearML.
Project description
Features
-
Graphs in ClearML - Display graphs for any Darknet compatible networks.
-
For classifiers:
- Accuracy (acc);
- Average loss (avg_loss);
- learning rate (lr).
-
For detectors:
- Mean average precision 50% (mAP@0.50);
- Precision;
- Recall;
- F1-score;
- Average loss (avg_loss);
- learning rate (lr).
-
-
Logs in ClearML - Displays logs from Darknet in ClearML.
-
Saving weights in ClearML - Ability to save weights in ClearML with flexible adjustment of parameters.
Available options:
- Darknet.is_save_weights - Ability to disable saving weights (for experiments).
- Darknet.save_weights_from_n_iterations - Ability to save weights from the start of N iterations. For example, N=10000.
- Darknet.save_weights_every_n_iterations - Ability to save weights every N iterations. For example, N=5000.
-
Additional function of data splitting - Allows you to divide the data set into selections.
📚 Click to see example
from clearml import Dataset from clearml_darknet.utils import split_dataset dataset_path = Dataset.get(dataset_name='dataset-example', dataset_project='Tests/darknet').get_local_copy() train, valid = split_dataset( dataset_path=dataset_path, ratio=0.7, shuffle=True )
Requirements
Installation
Before running, be sure to clone the Darknet repository and compile for the appropriate operating system.
$ pip install clearml-darknet-py
Examples
- An example of a detector training script - train_detector.py
- An example of a classificator training script - train_classifier.py
- Detector optimization script example - optimize_detector.py
For examples and use cases, check the examples folder.
License
Clearml Darknet is MIT licensed as listed in the LICENSE file.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for clearml_darknet_py-0.2.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 977792f1eb8d1ee85b0ad912395124a37076a833d9d217b21671b2d91236f636 |
|
MD5 | 15a35b2cd6da76cdab378be50f69702b |
|
BLAKE2b-256 | 2353f0af9591db162053c335cce17fd069d18781ebef5dec728ad4015078a16c |