cortenforge-training 0.6.0

Burn-based training and eval for LinearClassifier/MultiboxModel models in the CortenForge stack.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122

[[bin]]
name = "eval"
path = "src/bin/eval.rs"

[[bin]]
name = "train"
path = "src/bin/train.rs"

[dependencies.anyhow]
version = "1.0.100"

[dependencies.bincode]
default-features = false
features = ["alloc", "std"]
version = "2.0.1"

[dependencies.burn]
default-features = false
features = ["autodiff", "std"]
version = "0.19.1"

[dependencies.burn-ndarray]
default-features = false
optional = true
version = "0.19.1"

[dependencies.burn-wgpu]
default-features = true
optional = true
version = "0.19.1"

[dependencies.burn_dataset]
features = ["burn-runtime"]
package = "cortenforge-burn-dataset"
version = "0.6.0"

[dependencies.clap]
features = ["derive"]
version = "4.5.54"

[dependencies.data_contracts]
package = "cortenforge-data-contracts"
version = "0.6.0"

[dependencies.image]
default-features = false
features = ["png"]
version = "0.25.9"

[dependencies.models]
package = "cortenforge-models"
version = "0.6.0"

[dependencies.serde]
features = ["derive"]
version = "1.0.228"

[dependencies.serde_json]
version = "1.0.149"

[dev-dependencies.tempfile]
version = "3.24.0"

[features]
backend-ndarray = ["burn-ndarray", "burn/ndarray"]
backend-wgpu = ["burn-wgpu", "burn/wgpu"]
convolutional_detector = ["models/convolutional_detector"]
default = ["backend-ndarray", "linear_detector"]
linear_detector = ["models/linear_detector"]

[lib]
name = "training"
path = "src/lib.rs"

[package]
autobenches = false
autobins = false
autoexamples = false
autolib = false
autotests = false
build = false
categories = ["science", "computer-vision"]
description = "Burn-based training and eval for LinearClassifier/MultiboxModel models in the CortenForge stack."
documentation = "https://docs.rs/cortenforge-training"
edition = "2021"
homepage = "https://github.com/via-balaena/CortenForge"
keywords = ["burn", "training", "computer-vision", "detection"]
license = "Apache-2.0"
name = "cortenforge-training"
publish = true
readme = "README.md"
repository = "https://github.com/via-balaena/CortenForge"
rust-version = "1.75"
version = "0.6.0"

[package.metadata.docs.rs]
all-features = true
rustdoc-args = ["--cfg", "docsrs"]

[[test]]
name = "dataset_load"
path = "tests/dataset_load.rs"

[[test]]
name = "forward_bigdet"
path = "tests/forward_bigdet.rs"

[[test]]
name = "forward_shapes_bigdet"
path = "tests/forward_shapes_bigdet.rs"

[[test]]
name = "smoke_bigdet"
path = "tests/smoke_bigdet.rs"

[[test]]
name = "smoke_train"
path = "tests/smoke_train.rs"

[[test]]
name = "warehouse_manifest_smoke"
path = "tests/warehouse_manifest_smoke.rs"