Add initial model and its training results.
Browse files- .gitignore +1 -0
- README.md +60 -0
- models/colour-checker-detection-l-seg.pt +3 -0
- training/colour-checker-detection-l-seg/train/BoxF1_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/BoxPR_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/BoxP_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/BoxR_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/MaskF1_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/MaskPR_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/MaskP_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/MaskR_curve.png +0 -0
- training/colour-checker-detection-l-seg/train/args.yaml +101 -0
- training/colour-checker-detection-l-seg/train/confusion_matrix.png +0 -0
- training/colour-checker-detection-l-seg/train/confusion_matrix_normalized.png +0 -0
- training/colour-checker-detection-l-seg/train/events.out.tfevents.1704226217.30e1602f14b6.4717.0 +3 -0
- training/colour-checker-detection-l-seg/train/labels.jpg +0 -0
- training/colour-checker-detection-l-seg/train/labels_correlogram.jpg +0 -0
- training/colour-checker-detection-l-seg/train/results.csv +0 -0
- training/colour-checker-detection-l-seg/train/results.png +0 -0
- training/colour-checker-detection-l-seg/train/train_batch0.jpg +0 -0
- training/colour-checker-detection-l-seg/train/train_batch1.jpg +0 -0
- training/colour-checker-detection-l-seg/train/train_batch2.jpg +0 -0
- training/colour-checker-detection-l-seg/train/val_batch0_labels.jpg +0 -0
- training/colour-checker-detection-l-seg/train/val_batch0_pred.jpg +0 -0
- training/colour-checker-detection-l-seg/val/BoxF1_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/BoxPR_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/BoxP_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/BoxR_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/MaskF1_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/MaskPR_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/MaskP_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/MaskR_curve.png +0 -0
- training/colour-checker-detection-l-seg/val/confusion_matrix.png +0 -0
- training/colour-checker-detection-l-seg/val/confusion_matrix_normalized.png +0 -0
- training/colour-checker-detection-l-seg/val/val_batch0_labels.jpg +0 -0
- training/colour-checker-detection-l-seg/val/val_batch0_pred.jpg +0 -0
.gitignore
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
.DS_Store
|
README.md
CHANGED
|
@@ -1,3 +1,63 @@
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: cc-by-4.0
|
| 3 |
+
task_categories:
|
| 4 |
+
- object-detection
|
| 5 |
+
datasets:
|
| 6 |
+
- colour-science/colour-checker-detection-dataset
|
| 7 |
+
tags:
|
| 8 |
+
- color
|
| 9 |
+
- color-checker
|
| 10 |
+
- color-checker-detection
|
| 11 |
+
- color-science
|
| 12 |
+
- color-space
|
| 13 |
+
- color-spaces
|
| 14 |
+
- colorspace
|
| 15 |
+
- colorspaces
|
| 16 |
+
- colour
|
| 17 |
+
- colour-checker
|
| 18 |
+
- colour-checker-detection
|
| 19 |
+
- colour-science
|
| 20 |
+
- colour-space
|
| 21 |
+
- colour-spaces
|
| 22 |
+
- colourspace
|
| 23 |
+
- colourspaces
|
| 24 |
+
- image
|
| 25 |
+
- model
|
| 26 |
+
- models
|
| 27 |
+
- segmentation
|
| 28 |
+
- yolo
|
| 29 |
+
pretty_name: Colour - Checker Detection - Models
|
| 30 |
---
|
| 31 |
+
|
| 32 |
+
# Colour - Checker Detection - Models
|
| 33 |
+
|
| 34 |
+
Various [YOLOv8](https://github.com/ultralytics/ultralytics) colour rendition charts detection models.
|
| 35 |
+
|
| 36 |
+
The weights obtained by training YOLOv8 segmentation with the [colour-science/colour-checker-detection-model](https://huggingface.co/datasets/colour-science/colour-checker-detection-model) dataset are supporting colour rendition charts detection in the [Colour Checker Detection](https://github.com/colour-science/colour-checker-detection) Python package.
|
| 37 |
+
|
| 38 |
+
## Classes
|
| 39 |
+
|
| 40 |
+
- **ColorCheckerClassic24**: Calibrite / X-Rite ColorCheckerClassic 24
|
| 41 |
+
|
| 42 |
+
## Models
|
| 43 |
+
|
| 44 |
+
| **Name** | **Size** | **mAP50** | **mAP50-95** |
|
| 45 |
+
|-----------------------------------|----------|-----------|--------------|
|
| 46 |
+
| colour-checker-detection-l-seg.pt | 92.6 MB | 0.995 | 0.96868 |
|
| 47 |
+
|
| 48 |
+
# Contact & Social
|
| 49 |
+
|
| 50 |
+
The *Colour Developers* can be reached via different means:
|
| 51 |
+
|
| 52 |
+
- [Email](mailto:[email protected]>)
|
| 53 |
+
- [Facebook](https://www.facebook.com/python.colour.science>)
|
| 54 |
+
- [Github Discussions](https://github.com/colour-science/colour-checker-detection/discussions>)
|
| 55 |
+
- [Gitter](https://gitter.im/colour-science/colour>)
|
| 56 |
+
- [Twitter](https://twitter.com/colour_science>)
|
| 57 |
+
|
| 58 |
+
# About
|
| 59 |
+
|
| 60 |
+
**Colour - Checker Detection - Models** by Colour Developers \
|
| 61 |
+
Copyright 2024 Colour Developers – [mailto:[email protected]]([email protected]) \
|
| 62 |
+
This software is released under terms of CC-BY-4.0: https://creativecommons.org/licenses/by/4.0/ \
|
| 63 |
+
[https://huggingface.co/colour-science/colour-checker-detection-models](https://huggingface.co/colour-science/colour-checker-detection-models)
|
models/colour-checker-detection-l-seg.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:acc785646c0fd06fd0a65dff29b074bbd78bf543109692c9105ffaef43c9f251
|
| 3 |
+
size 92579635
|
training/colour-checker-detection-l-seg/train/BoxF1_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/BoxPR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/BoxP_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/BoxR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/MaskF1_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/MaskPR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/MaskP_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/MaskR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/train/args.yaml
ADDED
|
@@ -0,0 +1,101 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
task: segment
|
| 2 |
+
mode: train
|
| 3 |
+
model: yolov8l-seg.pt
|
| 4 |
+
data: /content/colour-checker-detection-dataset
|
| 5 |
+
epochs: 600
|
| 6 |
+
time: null
|
| 7 |
+
patience: 50
|
| 8 |
+
batch: 16
|
| 9 |
+
imgsz: 1280
|
| 10 |
+
save: true
|
| 11 |
+
save_period: -1
|
| 12 |
+
cache: false
|
| 13 |
+
device: 0
|
| 14 |
+
workers: 8
|
| 15 |
+
project: null
|
| 16 |
+
name: train3
|
| 17 |
+
exist_ok: false
|
| 18 |
+
pretrained: true
|
| 19 |
+
optimizer: auto
|
| 20 |
+
verbose: true
|
| 21 |
+
seed: 0
|
| 22 |
+
deterministic: true
|
| 23 |
+
single_cls: false
|
| 24 |
+
rect: false
|
| 25 |
+
cos_lr: false
|
| 26 |
+
close_mosaic: 10
|
| 27 |
+
resume: false
|
| 28 |
+
amp: true
|
| 29 |
+
fraction: 1.0
|
| 30 |
+
profile: false
|
| 31 |
+
freeze: null
|
| 32 |
+
overlap_mask: true
|
| 33 |
+
mask_ratio: 4
|
| 34 |
+
dropout: 0.0
|
| 35 |
+
val: true
|
| 36 |
+
split: val
|
| 37 |
+
save_json: false
|
| 38 |
+
save_hybrid: false
|
| 39 |
+
conf: null
|
| 40 |
+
iou: 0.7
|
| 41 |
+
max_det: 300
|
| 42 |
+
half: false
|
| 43 |
+
dnn: false
|
| 44 |
+
plots: true
|
| 45 |
+
source: null
|
| 46 |
+
vid_stride: 1
|
| 47 |
+
stream_buffer: false
|
| 48 |
+
visualize: false
|
| 49 |
+
augment: false
|
| 50 |
+
agnostic_nms: false
|
| 51 |
+
classes: null
|
| 52 |
+
retina_masks: false
|
| 53 |
+
embed: null
|
| 54 |
+
show: false
|
| 55 |
+
save_frames: false
|
| 56 |
+
save_txt: false
|
| 57 |
+
save_conf: false
|
| 58 |
+
save_crop: false
|
| 59 |
+
show_labels: true
|
| 60 |
+
show_conf: true
|
| 61 |
+
show_boxes: true
|
| 62 |
+
line_width: null
|
| 63 |
+
format: torchscript
|
| 64 |
+
keras: false
|
| 65 |
+
optimize: false
|
| 66 |
+
int8: false
|
| 67 |
+
dynamic: false
|
| 68 |
+
simplify: false
|
| 69 |
+
opset: null
|
| 70 |
+
workspace: 4
|
| 71 |
+
nms: false
|
| 72 |
+
lr0: 0.01
|
| 73 |
+
lrf: 0.01
|
| 74 |
+
momentum: 0.937
|
| 75 |
+
weight_decay: 0.0005
|
| 76 |
+
warmup_epochs: 3.0
|
| 77 |
+
warmup_momentum: 0.8
|
| 78 |
+
warmup_bias_lr: 0.1
|
| 79 |
+
box: 7.5
|
| 80 |
+
cls: 0.5
|
| 81 |
+
dfl: 1.5
|
| 82 |
+
pose: 12.0
|
| 83 |
+
kobj: 1.0
|
| 84 |
+
label_smoothing: 0.0
|
| 85 |
+
nbs: 64
|
| 86 |
+
hsv_h: 0.05
|
| 87 |
+
hsv_s: 0.5
|
| 88 |
+
hsv_v: 0.25
|
| 89 |
+
degrees: 22.5
|
| 90 |
+
translate: 0.1
|
| 91 |
+
scale: 0.25
|
| 92 |
+
shear: 11.25
|
| 93 |
+
perspective: 0.0
|
| 94 |
+
flipud: 0.25
|
| 95 |
+
fliplr: 0.25
|
| 96 |
+
mosaic: 0
|
| 97 |
+
mixup: 0.0
|
| 98 |
+
copy_paste: 0.0
|
| 99 |
+
cfg: null
|
| 100 |
+
tracker: botsort.yaml
|
| 101 |
+
save_dir: runs/segment/train
|
training/colour-checker-detection-l-seg/train/confusion_matrix.png
ADDED
|
training/colour-checker-detection-l-seg/train/confusion_matrix_normalized.png
ADDED
|
training/colour-checker-detection-l-seg/train/events.out.tfevents.1704226217.30e1602f14b6.4717.0
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:1b5ca15e752c14f64307b1b2dae70c8d887940d916aa212cbe238364aed8de82
|
| 3 |
+
size 1184505
|
training/colour-checker-detection-l-seg/train/labels.jpg
ADDED
|
training/colour-checker-detection-l-seg/train/labels_correlogram.jpg
ADDED
|
training/colour-checker-detection-l-seg/train/results.csv
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
training/colour-checker-detection-l-seg/train/results.png
ADDED
|
training/colour-checker-detection-l-seg/train/train_batch0.jpg
ADDED
|
training/colour-checker-detection-l-seg/train/train_batch1.jpg
ADDED
|
training/colour-checker-detection-l-seg/train/train_batch2.jpg
ADDED
|
training/colour-checker-detection-l-seg/train/val_batch0_labels.jpg
ADDED
|
training/colour-checker-detection-l-seg/train/val_batch0_pred.jpg
ADDED
|
training/colour-checker-detection-l-seg/val/BoxF1_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/BoxPR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/BoxP_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/BoxR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/MaskF1_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/MaskPR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/MaskP_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/MaskR_curve.png
ADDED
|
training/colour-checker-detection-l-seg/val/confusion_matrix.png
ADDED
|
training/colour-checker-detection-l-seg/val/confusion_matrix_normalized.png
ADDED
|
training/colour-checker-detection-l-seg/val/val_batch0_labels.jpg
ADDED
|
training/colour-checker-detection-l-seg/val/val_batch0_pred.jpg
ADDED
|