Picture using minio service

This commit is contained in:
zhouchengrong
2024-03-18 14:40:46 +08:00
parent 2f89e7432f
commit 909a441a86
2 changed files with 26 additions and 4 deletions

22
Dockerfile Normal file
View File

@@ -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"]

View File

@@ -11,12 +11,12 @@ def sim_score(a, b):
class FashionDataset(object): class FashionDataset(object):
# fashion_categories = json.load(open(r"config/fashion_categories.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")) fashion_categories_mapping = json.load(open(r"config/fashion_category_mapping.json", "r"))
# service debug # service debug
fashion_categories = json.load(open(r"service/outfit_matcher/config/fashion_categories.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")) # fashion_categories_mapping = json.load(open(r"service/outfit_matcher/config/fashion_category_mapping.json", "r"))
def __init__(self, item_metadata): def __init__(self, item_metadata):
self.item_metadat = item_metadata self.item_metadat = item_metadata