diff --git a/aipipeline/projects/bio/model/inference.py b/aipipeline/prediction/inference.py similarity index 98% rename from aipipeline/projects/bio/model/inference.py rename to aipipeline/prediction/inference.py index 4b35bae..ce63679 100644 --- a/aipipeline/projects/bio/model/inference.py +++ b/aipipeline/prediction/inference.py @@ -1,12 +1,11 @@ # aipipeline, Apache-2.0 license -# Filename: projects/bio/model/inference.py -# Description: Base class for detection models +# Filename: prediction/inference.py +# Description: model inference classes for projects import glob import logging import time -import uuid from datetime import datetime -from typing import List, Dict, Any +from typing import List, Any import numpy as np import requests @@ -27,7 +26,7 @@ logger.setLevel(logging.INFO) # and log to file now = datetime.now() -log_filename = f"bio_model_{now:%Y%m%d}.log" +log_filename = f"inference_{now:%Y%m%d}.log" handler = logging.FileHandler(log_filename, mode="w") handler.setFormatter(formatter) handler.setLevel(logging.DEBUG) diff --git a/aipipeline/projects/bio/process.py b/aipipeline/projects/bio/process.py index 068548a..5c79c74 100644 --- a/aipipeline/projects/bio/process.py +++ b/aipipeline/projects/bio/process.py @@ -1,8 +1,10 @@ -import logging -import os +# aipipeline, Apache-2.0 license +# Filename: projects/bio/process.py +# Description: Compute stats for downloaded datasets and save to a csv file from datetime import datetime - import dotenv +import logging +import os import redis from biotrack.tracker import BioTracker @@ -15,7 +17,7 @@ from aipipeline.projects.bio.core.predict import Predictor from aipipeline.projects.bio.core.video import VideoSource -from aipipeline.projects.bio.model.inference import FastAPIYV5, YV5 +from aipipeline.prediction.inference import FastAPIYV5, YV5 # Global variables idv = 1 # video index