fix  重写所有resize代码,mmcv替换为cv
This commit is contained in:
zhouchengrong
2024-07-12 11:32:43 +08:00
parent cb4b2b4eef
commit db354cc02a
3 changed files with 24 additions and 22 deletions

View File

@@ -1,14 +1,14 @@
#!/usr/bin/env python
# -*- coding: UTF-8 -*-
import logging
from pprint import pprint
import torch
import cv2
import mmcv
import numpy as np
import pandas as pd
from minio import Minio
import torch
import tritonclient.http as httpclient
from app.core.config import *
from app.schemas.attribute_retrieve import AttributeRecognitionModel
from app.service.utils.oss_client import oss_get_image
@@ -107,12 +107,8 @@ class AttributeRecognition:
@staticmethod
def preprocess(img):
img = mmcv.imread(img)
ori_shape = img.shape[:2]
img_scale = (224, 224)
scale_factor = []
img, x, y = mmcv.imresize(img, img_scale, return_scale=True)
scale_factor.append(x)
scale_factor.append(y)
img = cv2.resize(img, img_scale, interpolation=cv2.INTER_LINEAR)
img = mmcv.imnormalize(img, mean=np.array([123.675, 116.28, 103.53]), std=np.array([58.395, 57.12, 57.375]), to_rgb=True)
preprocessed_img = np.expand_dims(img.transpose(2, 0, 1), axis=0)
return preprocessed_img