|
|
|
|
|
|
|
|
|
|
|
|
|
|
export const MODEL_PATHS = {
|
|
|
classifier: './models/classifier_model_compressed/model.onnx',
|
|
|
poorGood: './models/poor_good_compressed/model.onnx',
|
|
|
grader: './models/grader_model_compressed/model.onnx',
|
|
|
yolo: './models/yolo-cropper/best.onnx'
|
|
|
};
|
|
|
|
|
|
export const DETECTION_CONFIG = {
|
|
|
yolo: {
|
|
|
inputSize: 640,
|
|
|
confThreshold: 0.6,
|
|
|
iouThreshold: 0.45,
|
|
|
padding: 0.25
|
|
|
},
|
|
|
siglip: {
|
|
|
inputSize: 224,
|
|
|
mean: [0.5, 0.5, 0.5],
|
|
|
std: [0.5, 0.5, 0.5]
|
|
|
}
|
|
|
};
|
|
|
|
|
|
export const LABEL_MAPPINGS = {
|
|
|
classifier: { 0: 'no', 1: 'yes' },
|
|
|
poorGood: { 0: 'good', 1: 'poor' },
|
|
|
grader: {}
|
|
|
};
|
|
|
|
|
|
export const ONNX_CONFIG = {
|
|
|
wasmPaths: 'https://cdn.jsdelivr.net/npm/onnxruntime-web@1.17.0/dist/',
|
|
|
numThreads: 4
|
|
|
};
|
|
|
|
|
|
export const ZOOM_CONFIG = {
|
|
|
min: 0.5,
|
|
|
max: 3,
|
|
|
step: 0.1,
|
|
|
default: 1
|
|
|
};
|
|
|
|