From 909a441a86b51184ec32e6856bee7fdda739dc89 Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Mon, 18 Mar 2024 14:40:46 +0800 Subject: [PATCH] Picture using minio service --- Dockerfile | 22 ++++++++++++++++++++++ app/service/outfit_matcher/dataset.py | 8 ++++---- 2 files changed, 26 insertions(+), 4 deletions(-) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..4d8ef96 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,22 @@ +FROM python:3.9 +ENV TZ=Asia/Shanghai +RUN apt update +RUN apt install -y vim +RUN apt install -y libgl1-mesa-glx +COPY ./requirements.txt /requirements.txt +RUN pip install --upgrade pip +RUN pip install -r requirements.txt +RUN pip install gunicorn +RUN pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 +RUN pip install mmcv==1.4.2 -f https://download.openmmlab.com/mmcv/dist/cu117/torch1.13/index.html + +WORKDIR /app +COPY . . +ENV FLASK_APP=manage.py +LABEL maintainer="zchengrong@yeah.net" \ + description="My Python 3.9 - trinity mixi " \ + version="1.0" \ + name="trinity_mixi" + + +CMD ["gunicorn", "-c", "gunicorn_config.py", "manage:app"] \ No newline at end of file diff --git a/app/service/outfit_matcher/dataset.py b/app/service/outfit_matcher/dataset.py index 21766ff..e440c50 100644 --- a/app/service/outfit_matcher/dataset.py +++ b/app/service/outfit_matcher/dataset.py @@ -11,12 +11,12 @@ def sim_score(a, b): class FashionDataset(object): - # fashion_categories = json.load(open(r"config/fashion_categories.json", "r")) - # fashion_categories_mapping = json.load(open(r"config/fashion_category_mapping.json", "r")) + fashion_categories = json.load(open(r"config/fashion_categories.json", "r")) + fashion_categories_mapping = json.load(open(r"config/fashion_category_mapping.json", "r")) # service debug - fashion_categories = json.load(open(r"service/outfit_matcher/config/fashion_categories.json", "r")) - fashion_categories_mapping = json.load(open(r"service/outfit_matcher/config/fashion_category_mapping.json", "r")) + # fashion_categories = json.load(open(r"service/outfit_matcher/config/fashion_categories.json", "r")) + # fashion_categories_mapping = json.load(open(r"service/outfit_matcher/config/fashion_category_mapping.json", "r")) def __init__(self, item_metadata): self.item_metadat = item_metadata