lightly-studio 0.3.1__py3-none-any.whl → 0.3.3__py3-none-any.whl

lightly_studio/dist_lightly_studio_view_app/_app/immutable/nodes/7.CF0EDTR6.js

@@ -1 +0,0 @@
-import"../chunks/CWj6FrbW.js";import{p as f,u as $,a as h,g as o,d as r}from"../chunks/l7KrR96u.js";import{s as S,a as i}from"../chunks/XO7A28GO.js";import{A as T}from"../chunks/CvOmgdoc.js";import{u as _}from"../chunks/vC4nQVEB.js";import"../chunks/CxtLVaYz.js";import"../chunks/D6su9Aln.js";import"../chunks/Bx1xMsFy.js";import"../chunks/Bsi3UGy5.js";import"../chunks/69_IOA4Y.js";import{u as G}from"../chunks/DIRAtgl0.js";import"../chunks/BJXwVxaE.js";function k(p,t){f(t,!0);const[e,m]=S(),n=()=>i(l,"$lastGridType",e),s=()=>i(d,"$sampleSize",e),{datasetId:a,sampleSize:d,selectedAnnotationFilterIds:c}=t.data,{lastGridType:l}=_(),u=r(()=>G({dataset_id:a})),g=r(()=>o(u).clearTagsSelected);$(()=>{n()!=="annotations"&&o(g)()}),T(p,{get itemHeight(){return s().height},get itemWidth(){return s().width},dataset_id:a,selectedAnnotationFilterIds:c}),h(),m()}export{k as component};

lightly_studio/dist_lightly_studio_view_app/_app/immutable/nodes/8.Cw30LEcV.js

@@ -1 +0,0 @@
-import{p as T,f as B,n as Et,a as j,w as bt,l as V,$ as ct,g as a,d,c as f,r as m,s as S,t as F,L as Dt,V as Tt,ax as gt}from"../chunks/l7KrR96u.js";import{c as jt}from"../chunks/BJXwVxaE.js";import"../chunks/CWj6FrbW.js";import{c as Z,a as _,t as k,n as Ht}from"../chunks/Bu7uvVrG.js";import{i as O,a as D,s as q}from"../chunks/XO7A28GO.js";import{e as ut}from"../chunks/DjfY96ND.js";import{I as Rt,s as J,K as Ut,L as at,M as Ft,N as Ot,O as zt,P as Gt,p as xt,F as Kt,e as Wt,l as qt,h as Vt,C as It,z as At,E as wt,u as Zt,d as Jt,S as Qt,Q as Yt,a as Xt}from"../chunks/CvOmgdoc.js";import{g as tt}from"../chunks/CxtLVaYz.js";import{u as St}from"../chunks/vC4nQVEB.js";import{f as ht,S as ta}from"../chunks/D5-A_Ffd.js";import{r as U}from"../chunks/H7C68rOM.js";import"../chunks/D6su9Aln.js";import"../chunks/Bx1xMsFy.js";import{p as z}from"../chunks/Bsi3UGy5.js";import"../chunks/69_IOA4Y.js";import{S as aa}from"../chunks/B2FVR0s0.js";import{S as na,B as ea,a as oa,b as X,c as lt,H as sa,d as dt,D as ra,e as ia,Z as la,g as da,u as ca}from"../chunks/nAHhluT7.js";import{S as ua}from"../chunks/BylOuP6i.js";import{s as va,r as ma,a as fa}from"../chunks/D98V7j6A.js";import{s as W}from"../chunks/D6RI2Zrd.js";import{C as _a}from"../chunks/CDnpyLsT.js";import{s as pa}from"../chunks/DIRAtgl0.js";function ga(v,t){T(t,!0);let r=ma(t,["$$slots","$$events","$$legacy"]);Rt(v,va({name:"square-dashed"},()=>r,{iconNode:[["path",{d:"M5 3a2 2 0 0 0-2 2"}],["path",{d:"M19 3a2 2 0 0 1 2 2"}],["path",{d:"M21 19a2 2 0 0 1-2 2"}],["path",{d:"M5 21a2 2 0 0 1-2-2"}],["path",{d:"M9 3h1"}],["path",{d:"M9 21h1"}],["path",{d:"M14 3h1"}],["path",{d:"M14 21h1"}],["path",{d:"M3 9v1"}],["path",{d:"M21 9v1"}],["path",{d:"M3 14v1"}],["path",{d:"M21 14v1"}]],children:(n,o)=>{var s=Z(),p=B(s);pa(p,()=>t.children??Et),_(n,s)},$$slots:{default:!0}})),j()}const ha=async({dataset_id:v,cursor:t=0,limit:r=10,annotation_label_ids:i,tag_ids:n})=>{const o={data:[],error:void 0};try{const s=await jt.GET("/api/datasets/{dataset_id}/annotations",{params:{path:{dataset_id:v},query:{annotation_label_ids:i,tag_ids:n,cursor:t,limit:r}}});if(s.error)throw new Error(JSON.stringify(s.error,null,2));if(!s.data)throw new Error("No data");o.data=s.data.data}catch(s){o.error="Error loading annotations: "+String(s)}return o},ba=({annotationIndex:v,...t})=>{const r=bt({isLoading:!1});return(async()=>{r.update(n=>({...n,isLoading:!0}));try{const n=await ha({cursor:v>0?v-1:0,limit:3,...t}),o=n.data.findIndex(e=>e.annotation_id===t.currentAnnotationId),s=n.data[o+1],p=n.data[o-1];r.update(e=>({...e,isLoading:!1,error:n.error?String(n.error):void 0,data:n.error?void 0:{annotationNext:s,annotationPrevious:p}}))}catch(n){r.update(o=>({...o,isLoading:!1,error:String(n)}))}})(),r},xa=async({parent:v,params:{dataset_id:t,annotationId:r,annotationIndex:i}})=>{const{annotationsSelectedTagsIds:n,annotationsSelectedAnnotationLabelsIds:o,dataset:s}=await v(),p=parseInt(i,10),e=bt();if(!s)throw new Error("Dataset data not found");return ba({annotationIndex:p,dataset_id:t,currentAnnotationId:r,annotation_label_ids:Array.from(V(o)),tag_ids:Array.from(V(n))}).subscribe(({data:L})=>{e.set(L)}),{annotationAdjacents:e,annotationIndex:p,annotationId:r,dataset:s}},gn=Object.freeze(Object.defineProperty({__proto__:null,load:xa},Symbol.toStringTag,{value:"Module"}));function Ia(v,t){T(t,!0);const[r,i]=q(),n=()=>D(a(s),"$annotationAdjacents",r),o=d(()=>z.data.annotationIndex),s=d(()=>z.data.annotationAdjacents),p=()=>{n().annotationNext&&tt(U.toSampleWithAnnotation({datasetId:n().annotationNext.dataset_id,sampleId:n().annotationNext.sample_id,annotationId:n().annotationNext.annotation_id,annotationIndex:a(o)+1}))},e=()=>{n().annotationPrevious&&tt(U.toSampleWithAnnotation({datasetId:n().annotationPrevious.dataset_id,sampleId:n().annotationPrevious.sample_id,annotationId:n().annotationPrevious.annotation_id,annotationIndex:a(o)-1}))},L=g=>{switch(g.key){case"ArrowRight":p();break;case"ArrowLeft":e();break}};var x=Z();ut("keydown",ct,L);var $=B(x);{var I=g=>{const P=d(()=>!!n().annotationPrevious),y=d(()=>!!n().annotationNext);ua(g,{get hasPrevious(){return a(P)},get hasNext(){return a(y)},onPrevious:e,onNext:p})};O($,g=>{n()&&g(I)})}_(v,x),j(),i()}const Aa={object_detection:"Object Detection",bbox:"Bounding Box",instance_segmentation:"Instance Segmentation",semantic_segmentation:"Semantic Segmentation",classification:"Classification"};var wa=k('<span class="text-sm"> </span> <span class="break-all text-sm"> </span>',1);function Sa(v,t){var r=wa(),i=B(r),n=f(i,!0);m(i);var o=S(i,2),s=f(o,!0);m(o),F(()=>{W(n,t.label),J(o,"data-testid",`annotation-metadata-${t.id}`),W(s,t.value)}),_(v,r)}var $a=k('<span class="break-all text-sm"> </span>'),ya=k('<span class="text-sm"> </span> <!>',1);function ka(v,t){T(t,!0);const[r,i]=q(),n=()=>D(s,"$result",r),o=()=>D(t.isEditingMode,"$isEditingMode",r),s=Ut(),{updateAnnotation:p}=at({datasetId:t.datasetId,annotationId:t.annotationId,onUpdate:t.onUpdate}),e=d(()=>zt(n().data||[])),L=d(()=>{const A=a(e).find(c=>c.value===t.value);return A||{value:t.value,label:t.value}});var x=ya(),$=B(x),I=f($,!0);m($);var g=S($,2);{var P=A=>{Ft(A,{get items(){return a(e)},get value(){return a(L)},name:"annotation-label",placeholder:"Select a label",onSelect:l=>{p({annotation_id:t.annotationId,dataset_id:t.datasetId,label_name:l.value})},notFound:(l,u)=>{let h=()=>u==null?void 0:u().inputValue;Ot(l,{get label(){return h()}})},$$slots:{notFound:!0}})},y=A=>{var c=$a();J(c,"data-testid","annotation-metadata-label");var l=f(c,!0);m(c),F(()=>W(l,t.value)),_(A,c)};O(g,A=>{o()?A(P):A(y,!1)})}F(()=>W(I,t.label)),_(v,x),j(),i()}var Pa=k('<div class="flex flex-col gap-4"><div class="grid grid-cols-[6rem_1fr] gap-y-3 text-diffuse-foreground"></div></div>');function Ma(v,t){T(t,!0);const[r,i]=q(),n=()=>D(a(p),"$annotationResp",r),{datasetId:o}=z.data,s=d(()=>at({datasetId:o,annotationId:t.annotationId})),p=d(()=>a(s).annotation);let e=d(()=>n().data);const L=d(qt),x=d(()=>a(L).taskMap),$=d(()=>a(e)&&a(e).annotation_task_id&&V(a(x)).has(a(e).annotation_task_id)?V(a(x)).get(a(e).annotation_task_id).is_prediction:!1),I=d(()=>{var l,u;if(!a(e))return[];let c=[{id:"label",label:"Label:",value:(u=(l=a(e))==null?void 0:l.annotation_label)==null?void 0:u.annotation_label_name},{id:"type",label:"Type:",value:Aa[a(e).annotation_type]||"Unknown"},{id:"status",label:"Status:",value:a($)?"Prediction":"Ground Truth"}];if(a(e)&&!Gt(a(e))){const{height:h,width:M}=xt(a(e));c=[{id:"height",label:"Height:",value:ht(Math.round(h))+"px"},{id:"width",label:"Width:",value:ht(Math.round(M))+"px"},...c]}return c}),{isEditingMode:g}=z.data.globalStorage;var P=Z(),y=B(P);{var A=c=>{Kt(c,{title:"Annotation details",children:(l,u)=>{var h=Pa(),M=f(h);Wt(M,21,()=>a(I),({label:N,value:b,id:C})=>N,(N,b)=>{let C=()=>a(b).label,E=()=>a(b).value,H=()=>a(b).id;var G=Z(),nt=B(G);{var Q=w=>{ka(w,{get onUpdate(){return t.onUpdate},get annotationId(){return t.annotationId},datasetId:o,get label(){return C()},get value(){return E()},isEditingMode:g})},et=w=>{Sa(w,{get label(){return C()},get id(){return H()},get value(){return E()}})};O(nt,w=>{H()==="label"&&g?w(Q):w(et,!1)})}_(N,G)}),m(M),m(h),_(l,h)}})};O(y,c=>{a(I).length>0&&c(A)})}_(v,P),j(),i()}var Na=k('<!> <h2 class="py-2 text-lg font-semibold"><a>Check all sample annotations</a></h2>',1),Ca=k('<div class="flex h-full w-full items-center justify-center"><!></div>'),Ba=k('<div class="flex h-full min-h-0 flex-col space-y-4 overflow-hidden dark:[color-scheme:dark]"><!> <!></div>');function La(v,t){T(t,!0);const{datasetId:r}=z.data;let i=Tt(void 0);at({datasetId:r,annotationId:t.annotationId}).annotation.subscribe(e=>{e.isSuccess&&Dt(i,fa(e.data.sample))});const o=Vt(),[s,p]=o;It(v,{className:"h-full",children:(e,L)=>{At(e,{className:"h-full flex flex-col",children:(x,$)=>{var I=Ba(),g=f(I);Ma(g,{get annotationId(){return t.annotationId},get onUpdate(){return t.onUpdate}});var P=S(g,2);{var y=c=>{var l=Na(),u=B(l);na(u,{get sample(){return a(i)},showCustomMetadata:!1});var h=S(u,2),M=f(h);m(h),F(N=>J(M,"href",N),[()=>s(U.toSample({sampleId:a(i).sample_id,datasetId:a(i).dataset_id}),void 0)]),_(c,l)},A=c=>{var l=Ca(),u=f(l);wt(u,{size:"large",align:"center"}),m(l),_(c,l)};O(P,c=>{a(i)?c(y):c(A,!1)})}m(I),_(x,I)},$$slots:{default:!0}})},$$slots:{default:!0}}),j()}var Ea=k('<!> <span class="hidden sm:inline">Home</span>',1),Da=k('<!> <span class="max-w-[150px] truncate"> </span>',1),Ta=k('<!> <span class="hidden sm:inline">Annotations</span>',1),ja=k('<!> <span class="max-w-[200px] truncate"> </span>',1),Ha=k("<!> <!> <!> <!> <!> <!> <!>",1);function Ra(v,t){T(t,!0);const[r,i]=q(),n=()=>D(o,"$annotationsTotalCount",r),{annotationsTotalCount:o}=St();ea(v,{class:"mb-2",children:(s,p)=>{oa(s,{children:(e,L)=>{var x=Ha(),$=B(x);X($,{children:(l,u)=>{const h=d(()=>U.toHome());lt(l,{get href(){return a(h)},class:"flex items-center gap-2",children:(M,N)=>{var b=Ea(),C=B(b);sa(C,{class:"h-4 w-4"}),gt(2),_(M,b)},$$slots:{default:!0}})},$$slots:{default:!0}});var I=S($,2);dt(I,{});var g=S(I,2);X(g,{children:(l,u)=>{const h=d(()=>U.toAnnotations(t.dataset.dataset_id));lt(l,{get href(){return a(h)},class:"flex items-center gap-2",get title(){return t.dataset.directory},children:(M,N)=>{var b=Da(),C=B(b);ra(C,{class:"h-4 w-4"});var E=S(C,2),H=f(E,!0);m(E),F(()=>{J(E,"title",t.dataset.directory),W(H,t.dataset.name)}),_(M,b)},$$slots:{default:!0}})},$$slots:{default:!0}});var P=S(g,2);dt(P,{});var y=S(P,2);X(y,{children:(l,u)=>{const h=d(()=>U.toAnnotations(t.dataset.dataset_id));lt(l,{get href(){return a(h)},class:"flex items-center gap-2",children:(M,N)=>{var b=Ta(),C=B(b);_a(C,{class:"h-4 w-4"}),gt(2),_(M,b)},$$slots:{default:!0}})},$$slots:{default:!0}});var A=S(y,2);dt(A,{});var c=S(A,2);X(c,{children:(l,u)=>{ia(l,{class:"flex items-center gap-2",children:(h,M)=>{var N=ja(),b=B(N);ga(b,{class:"h-4 w-4"});var C=S(b,2),E=f(C);m(C),F(()=>W(E,`Annotation ${t.annotationIndex+1} of ${n()??""}`)),_(h,N)},$$slots:{default:!0}})},$$slots:{default:!0}}),_(e,x)},$$slots:{default:!0}})},$$slots:{default:!0}}),j(),i()}var Ua=Ht("<image></image><g><!></g>",1),Fa=k('<div class="h-full w-full overflow-hidden"><div class="sample relative h-full w-full"><div class="absolute right-4 top-2 z-30"><!></div> <!> <!></div></div>'),Oa=k('<div class="flex h-full w-full items-center justify-center"><!></div>'),za=k('<div class="flex h-full w-full flex-col space-y-4"><div class="flex w-full items-center"><!></div> <!> <div class="flex min-h-0 flex-1 gap-4"><div class="flex-1"><!></div> <div class="relative w-[375px]"><!></div></div></div>');function Ga(v,t){T(t,!0);const[r,i]=q(),n=()=>D(a(l),"$annotationResp",r),o=()=>D(e,"$selectedSampleAnnotationCropIds",r),s=()=>D(L,"$isHidden",r),{toggleSampleAnnotationCropSelection:p,selectedSampleAnnotationCropIds:e}=St(),{isHidden:L,handleKeyEvent:x}=Zt(),{settingsStore:$}=Jt(),I=" ",g=V($).key_go_back,P=()=>{var w;(w=t.sample)!=null&&w.dataset_id?tt(U.toAnnotations(t.sample.dataset_id)):tt("/")},y=w=>{switch(w.key){case g:P();break;case I:w.preventDefault(),p(t.annotationId);break}x(w)},A=z.data.datasetId,c=d(()=>at({datasetId:A,annotationId:t.annotationId})),l=d(()=>a(c).annotation);let u=d(()=>n().data),h=d(()=>{var w,ot;return da(((ot=(w=a(u))==null?void 0:w.sample)==null?void 0:ot.sample_id)||"")}),M=d(()=>a(u)?xt(a(u)):void 0);var N=za();ut("keydown",ct,y),ut("keyup",ct,x);var b=f(N),C=f(b);Ra(C,{get dataset(){return t.dataset},get annotationIndex(){return t.annotationIndex}}),m(b);var E=S(b,2);ta(E,{class:"mb-4 bg-border-hard"});var H=S(E,2),G=f(H),nt=f(G);It(nt,{className:"h-full",children:(w,ot)=>{At(w,{className:"h-full",children:($t,Wa)=>{var vt=Z(),yt=B(vt);{var kt=K=>{var R=Fa(),Y=f(R),st=f(Y),Mt=f(st);const Nt=d(()=>o().has(t.annotationId));Qt(Mt,{onSelect:()=>p(t.annotationId),get isSelected(){return a(Nt)}}),m(st);var mt=S(st,2);Ia(mt,{});var Ct=S(mt,2);la(Ct,{get width(){return a(u).sample.width},get height(){return a(u).sample.height},get boundingBox(){return a(M)},children:(Bt,qa)=>{var ft=Ua(),_t=B(ft),rt=S(_t);let pt;var Lt=f(rt);Yt(Lt,()=>a(u).annotation_id,it=>{aa(it,{get annotation(){return a(u)},showLabel:!0,get imageWidth(){return a(u).sample.width}})}),m(rt),F(it=>{J(_t,"href",a(h)),pt=Xt(rt,0,"",null,pt,it)},[()=>({invisible:s()})]),_(Bt,ft)},$$slots:{default:!0}}),m(Y),m(R),_(K,R)},Pt=K=>{var R=Oa(),Y=f(R);wt(Y,{size:"large",align:"center"}),m(R),_(K,R)};O(yt,K=>{a(u)?K(kt):K(Pt,!1)})}_($t,vt)},$$slots:{default:!0}})},$$slots:{default:!0}}),m(G);var Q=S(G,2),et=f(Q);La(et,{get annotationId(){return t.annotationId}}),m(Q),m(H),m(N),_(v,N),j(),i()}var Ka=k('<div class="flex h-full w-full space-x-4 px-4 pb-4" data-testid="annotation-details"><div class="h-full w-full space-y-6 rounded-[1vw] bg-card p-4"><!></div></div>');function hn(v,t){T(t,!0);const[r,i]=q(),n=()=>D(a(x),"$sample",r),o=d(()=>t.data.annotationId),s=d(()=>t.data.dataset),p=d(()=>t.data.annotationIndex),{sampleId:e}=z.params,L=d(()=>ca({sampleId:e,datasetId:a(s).dataset_id})),x=d(()=>a(L).sample);var $=Ka(),I=f($),g=f(I);{var P=y=>{Ga(y,{get annotationId(){return a(o)},get annotationIndex(){return a(p)},get dataset(){return a(s)},get sample(){return n().data}})};O(g,y=>{n().data&&a(o)&&a(s)&&y(P)})}m(I),m($),_(v,$),j(),i()}export{hn as component,gn as universal};

lightly_studio/metrics/detection/map.py

@@ -1,268 +0,0 @@
-"""Implementation for calculating Mean Average Precision (MAP)."""
-
-from __future__ import annotations
-
-import math
-from collections import defaultdict
-from collections.abc import Sequence
-from uuid import UUID
-
-import torch
-from pydantic import BaseModel
-from torch import Tensor
-from torchmetrics.detection.mean_ap import MeanAveragePrecision
-
-from lightly_studio.models.annotation.annotation_base import AnnotationBaseTable
-
-
-class DetectionMetricsMAP(BaseModel):
-    """Response for computing the MAP detection metric."""
-
-    name: str
-    map: float
-    map_small: float
-    map_medium: float
-    map_large: float
-    mar_1: float
-    mar_10: float
-    mar_100: float
-    mar_small: float
-    mar_medium: float
-    mar_large: float
-    map_50: float
-    map_75: float
-    map_per_class: dict[str, float] | None = None
-    mar_100_per_class: dict[str, float] | None = None
-    classes: list[int]
-
-    @property
-    def value(self) -> float:
-        """Backwards compatibility alias for map."""
-        return self.map
-
-    @property
-    def per_class(self) -> dict[str, float] | None:
-        """Backwards compatibility alias for map_per_class."""
-        return self.map_per_class
-
-
-def calculate_map_metric(  # noqa: C901
-    pred_annotations: Sequence[AnnotationBaseTable],
-    gt_annotations: Sequence[AnnotationBaseTable],
-) -> DetectionMetricsMAP:
-    """Calculate the Mean Average Precision (MAP) metric."""
-    if not gt_annotations or not pred_annotations:
-        return DetectionMetricsMAP(
-            name="mAP@.5:.95",
-            map=0.0,
-            map_small=0.0,
-            map_medium=0.0,
-            map_large=0.0,
-            mar_1=0.0,
-            mar_10=0.0,
-            mar_100=0.0,
-            mar_small=0.0,
-            mar_medium=0.0,
-            mar_large=0.0,
-            map_50=-1.0,
-            map_75=-1.0,
-            map_per_class=None,
-            mar_100_per_class=None,
-            classes=[],
-        )
-    uuid_to_int_label: dict[UUID, int] = {}
-
-    # Map sample_id to annotations.
-    sample_id_to_annotations = _group_by_sample_id(
-        pred_annotations=pred_annotations,
-        gt_annotations=gt_annotations,
-    )
-
-    # Create the MeanAveragePrecision object.
-    map_metric = MeanAveragePrecision(
-        box_format="xywh",
-        iou_thresholds=None,
-        class_metrics=True,
-        average="macro",
-        backend="faster_coco_eval",
-    )
-
-    # Compute MAP updating the object sample by sample.
-    for _sample_id, (
-        sample_pred_annotations,
-        sample_gt_annotations,
-    ) in sample_id_to_annotations.items():
-        # Convert to torchmetrics format.
-        prediction_tm = _convert_to_torchmetrics(
-            annotations=sample_pred_annotations,
-            is_prediction=True,
-            uuid_to_int_label_map=uuid_to_int_label,
-        )
-        ground_truth_tm = _convert_to_torchmetrics(
-            annotations=sample_gt_annotations,
-            is_prediction=False,
-            uuid_to_int_label_map=uuid_to_int_label,
-        )
-
-        # Update the metric.
-        map_metric.update(preds=[prediction_tm], target=[ground_truth_tm])
-
-    # Compute the final results.
-    results = map_metric.compute()
-    # Invert label map to recover original UUIDs
-    int_to_uuid = {v: k for k, v in uuid_to_int_label.items()}
-
-    # Helper to convert Tensor to Python list
-    def to_list(t: Tensor) -> list[float]:
-        if t.dim() == 0:
-            return [t.item()]
-        return t.tolist()
-
-    # Scalar metrics
-    map_val = results["map"].item()
-    map_small = results["map_small"].item()
-    map_medium = results["map_medium"].item()
-    map_large = results["map_large"].item()
-    mar_1 = results["mar_1"].item()
-    mar_10 = results["mar_10"].item()
-    mar_100 = results["mar_100"].item()
-    mar_small = results["mar_small"].item()
-    mar_medium = results["mar_medium"].item()
-    mar_large = results["mar_large"].item()
-
-    # IoU-specific metrics
-    map_50 = results.get("map_50")
-    map_50 = map_50.item() if isinstance(map_50, Tensor) and map_50.dim() == 0 else -1.0
-    map_75 = results.get("map_75")
-    map_75 = map_75.item() if isinstance(map_75, Tensor) and map_75.dim() == 0 else -1.0
-
-    # Per-class metrics
-    raw_map_pc = results.get("map_per_class")
-    per_class_dict: dict[str, float] | None = None
-    if raw_map_pc is not None:
-        pc_list = to_list(raw_map_pc)
-        per_class_dict = {}
-        for idx, val in enumerate(pc_list):
-            if not math.isnan(val):
-                per_class_dict[str(int_to_uuid[idx])] = val
-
-    # Per-class recall at 100 detections
-    raw_mar100_pc = results.get("mar_100_per_class")
-    mar_100_pc_dict: dict[str, float] | None = None
-    if raw_mar100_pc is not None:
-        m100_list = to_list(raw_mar100_pc)
-        mar_100_pc_dict = {}
-        for idx, val in enumerate(m100_list):
-            if not math.isnan(val):
-                mar_100_pc_dict[str(int_to_uuid[idx])] = val
-
-    # Observed classes
-    raw_classes = results.get("classes")
-    classes_list_output: list[int] = []
-    if raw_classes is not None:
-        classes_list_output = [int(x) for x in to_list(raw_classes)]
-
-    return DetectionMetricsMAP(
-        name="mAP@.5:.95",
-        map=map_val,
-        map_small=map_small,
-        map_medium=map_medium,
-        map_large=map_large,
-        mar_1=mar_1,
-        mar_10=mar_10,
-        mar_100=mar_100,
-        mar_small=mar_small,
-        mar_medium=mar_medium,
-        mar_large=mar_large,
-        map_50=map_50,
-        map_75=map_75,
-        map_per_class=per_class_dict,
-        mar_100_per_class=mar_100_pc_dict,
-        classes=classes_list_output,
-    )
-
-
-def _group_by_sample_id(
-    pred_annotations: Sequence[AnnotationBaseTable],
-    gt_annotations: Sequence[AnnotationBaseTable],
-) -> dict[
-    UUID,
-    tuple[list[AnnotationBaseTable], list[AnnotationBaseTable]],
-]:
-    """Group prediction and ground truth annotations by sample_id.
-
-    Returns a dictionary with sample_id as key and a tuple of
-    (list of prediction annotations, list of ground truth annotations) as value.
-    """
-    sample_id_to_annotations: defaultdict[
-        UUID,
-        tuple[list[AnnotationBaseTable], list[AnnotationBaseTable]],
-    ] = defaultdict(lambda: ([], []))
-    for ann in pred_annotations:
-        sample_id_to_annotations[ann.sample_id][0].append(ann)
-    for ann in gt_annotations:
-        sample_id_to_annotations[ann.sample_id][1].append(ann)
-    return sample_id_to_annotations
-
-
-def _convert_to_torchmetrics(
-    annotations: Sequence[AnnotationBaseTable],
-    is_prediction: bool,
-    uuid_to_int_label_map: dict[UUID, int],
-) -> dict[str, Tensor]:
-    """Convert annotations to torchmetrics format.
-
-    Args:
-        annotations: List of bounding box annotations.
-        is_prediction: Whether the annotations are predictions.
-        uuid_to_int_label_map: Map from UUID to integer label.
-
-    Returns:
-        Dictionary in torchmetrics format. For predictions the keys are
-        `boxes`, `scores`, `labels`. For ground truth they are `boxes`,
-        `labels`.
-
-    """
-    if not annotations:
-        empty_boxes = torch.empty((0, 4), dtype=torch.float32)
-        empty_labels = torch.empty((0,), dtype=torch.int64)
-        if is_prediction:
-            return {
-                "boxes": empty_boxes,
-                "scores": torch.empty((0,), dtype=torch.float32),
-                "labels": empty_labels,
-            }
-        return {"boxes": empty_boxes, "labels": empty_labels}
-
-    boxes = torch.tensor(
-        [
-            [
-                a.object_detection_details.x,
-                a.object_detection_details.y,
-                a.object_detection_details.width,
-                a.object_detection_details.height,
-            ]
-            for a in annotations
-            if a.object_detection_details is not None
-        ],
-        dtype=torch.float32,
-    )
-
-    mapped_labels = []
-    # Use the passed-in map instead of a global one
-    for uuid_label in (a.annotation_label_id for a in annotations):
-        if uuid_label not in uuid_to_int_label_map:
-            # Assign the next available integer ID based on the current map size
-            uuid_to_int_label_map[uuid_label] = len(uuid_to_int_label_map)
-        mapped_labels.append(uuid_to_int_label_map[uuid_label])
-
-    labels = torch.tensor(mapped_labels, dtype=torch.int64)
-
-    if is_prediction:
-        scores = torch.tensor([a.confidence for a in annotations], dtype=torch.float32)
-        return {
-            "boxes": boxes,
-            "scores": scores,
-            "labels": labels,
-        }
-    return {"boxes": boxes, "labels": labels}

lightly_studio/models/annotation_task.py

@@ -1,28 +0,0 @@
-"""This module defines the AnnotationTask model."""
-
-from datetime import datetime
-from enum import Enum
-from uuid import UUID, uuid4
-
-from sqlmodel import Field, SQLModel
-
-
-class AnnotationType(str, Enum):
-    """The type of annotation task."""
-
-    BBOX = "bbox"
-    CLASSIFICATION = "classification"
-    SEMANTIC_SEGMENTATION = "semantic_segmentation"
-    INSTANCE_SEGMENTATION = "instance_segmentation"
-    OBJECT_DETECTION = "object_detection"
-
-
-class AnnotationTaskTable(SQLModel, table=True):
-    """The annotation task model."""
-
-    __tablename__ = "annotation_tasks"
-    annotation_task_id: UUID = Field(default_factory=uuid4, primary_key=True)
-    name: str
-    created_at: datetime = Field(default_factory=datetime.utcnow)
-    annotation_type: AnnotationType
-    is_prediction: bool

lightly_studio/resolvers/annotation_resolver/create.py

@@ -1,19 +0,0 @@
-"""Handler for database operations related to annotations."""
-
-from __future__ import annotations
-
-from sqlmodel import Session
-
-from lightly_studio.models.annotation.annotation_base import (
-    AnnotationBaseTable,
-    AnnotationCreate,
-)
-
-
-def create(session: Session, annotation: AnnotationCreate) -> AnnotationBaseTable:
-    """Create a new annotation in the database."""
-    db_annotation = AnnotationBaseTable.model_validate(annotation)
-    session.add(db_annotation)
-    session.commit()
-    session.refresh(db_annotation)
-    return db_annotation

lightly_studio/resolvers/annotation_task_resolver.py

@@ -1,31 +0,0 @@
-"""Resolver for annotation tasks."""
-
-from typing import List, Optional, Sequence
-from uuid import UUID
-
-from sqlmodel import Session, col, select
-
-from lightly_studio.models.annotation_task import AnnotationTaskTable
-
-
-def create(session: Session, annotation_task: AnnotationTaskTable) -> AnnotationTaskTable:
-    """Create a new annotation task."""
-    session.add(annotation_task)
-    session.commit()
-    session.refresh(annotation_task)
-    return annotation_task
-
-
-def get_by_id(session: Session, annotation_task_id: UUID) -> Optional[AnnotationTaskTable]:
-    """Get an annotation task by ID."""
-    statement = select(AnnotationTaskTable).where(
-        AnnotationTaskTable.annotation_task_id == annotation_task_id
-    )
-    return session.exec(statement).first()
-
-
-def get_all(session: Session) -> List[AnnotationTaskTable]:
-    """Get all annotation tasks."""
-    statement = select(AnnotationTaskTable).order_by(col(AnnotationTaskTable.created_at).asc())
-    results: Sequence[AnnotationTaskTable] = session.exec(statement).all()
-    return list(results)