This commit is contained in:
TannedCung 2024-06-11 14:47:09 +07:00
parent 9ff1174b89
commit 65479e6844
4 changed files with 44 additions and 58 deletions

File diff suppressed because one or more lines are too long

View File

@ -144,6 +144,7 @@ def create_accuracy_report(report_id, **kwargs):
request.is_required = request.raw_accuracy < settings.NEED_REVIEW if isinstance(request.raw_accuracy, float) else False # No need to review by default
request.save()
number_images += request_att["total_images"]
number_bad_images += request_att["bad_images"]
bad_image_list += request_att["bad_image_list"]

View File

@ -125,6 +125,7 @@ class ReportAccumulateByRequest:
@staticmethod
def update_total(total, report_file):
# Update total = update month
if report_file.bad_image_reason not in settings.ACC_EXCLUDE_RESEASONS:
total["images_quality"]["successful"] += 1 if not report_file.is_bad_image else 0
total["images_quality"]["bad"] += 1 if report_file.is_bad_image else 0

View File

@ -17,25 +17,26 @@ services:
- ctel-sbt
privileged: true
image: sidp/cope2n-ai-fi-sbt:latest
# runtime: nvidia
environment:
- PYTHONPATH=${PYTHONPATH}:/workspace/cope2n-ai-fi # For import module
- CELERY_BROKER=amqp://${RABBITMQ_DEFAULT_USER}:${RABBITMQ_DEFAULT_PASS}@rabbitmq-sbt:5672
- CUDA_VISIBLE_DEVICES=0
# - CUDA_VISIBLE_DEVICES=0
volumes:
- ./cope2n-ai-fi:/workspace/cope2n-ai-fi # for dev container only
working_dir: /workspace/cope2n-ai-fi
# deploy:
# resources:
# reservations:
# devices:
# - driver: nvidia
# count: 1
# capabilities: [gpu]
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: 1
capabilities: [gpu]
# command: bash -c "tail -f > /dev/null"
command: bash run.sh
deploy:
mode: replicated
replicas: 1
# deploy:
# mode: replicated
# replicas: 1
# Back-end services
be-ctel-sbt:
build:
@ -195,8 +196,8 @@ services:
- POSTGRES_USER=${DB_USER}
- POSTGRES_PASSWORD=${DB_PASSWORD}
- POSTGRES_DB=${DB_SCHEMA}
ports:
- 54321:5432
# ports:
# - 54321:5432
rabbitmq-sbt:
mem_reservation: 600m