long-utils 0.6

A utility library for everyday use.

安装:

pip install long-utils

图像

图中找图

from long_utils.image.search import ImgFindImg

img = ImgFindImg(img_source='static/big.png', img_search='static/small.png')
img.find_coord(is_debug=True)

# 或者只拿取小图在原图中的坐标rectangle, 小图在原图中的大小result，confidence表示准确率
res = img.find_coord()
# [{'result': (539.0, 21.5), 'rectangle': ((524, 10), (524, 33), (554, 10), (554, 33)), 'confidence': 1.0}]
print(res)

其他

日志输出

from long_utils import Log

logger = Log(show_line=True, show_file_name=True, show_time=True)

logger.error(message="xxxx")
logger.success(message="xxxx")
logger.warning(message="xxxx")
logger.info(message="xxxx")
logger.debug(message="xxxx")

logger.success_bg('xxxx')
logger.error_bg('xxxx')
logger.warning_bg('xxxx')
logger.debug_bg('xxxx')
logger.info_bg('xxxx')
logger.color('xxxx', fg='blue', bg='black')

对象存储OSS

配置

from long_utils.storage.config import SSconfig
from long_utils.storage.aliyun.oss import AliyumOss


access_key = 'xxx'
access_key_secret = 'xxx'
endpoint = 'https://oss-cn-shanghai.aliyuncs.com'
object_name = 'xxx'
ss_config = SSconfig(
    access_key_id=access_key,
    access_key_secret=access_key_secret,
    bucket_url=endpoint,
    bucket_name=object_name
)
endpoint_host, endpoint_sec = endpoint.split('//')
object_image_url = f"{endpoint_host}//{object_name}.{endpoint_sec}"
aliyun_client = AliyumOss(
    config=ss_config,
    object_image_url=object_image_url
)

分片下载文件到本地

aliyun_client.multipart_upload_file(key='xxx.pth', local_file='xxx.pth')

如果需要知道下载进度,可以:

def progress_callback(block_size, total_size):
    print(f"正在下载: {block_size / total_size * 100:.2f}%")

aliyun_client.multipart_upload_file(key='xxx.pth', local_file='xxx.pth', progress_callback=progress_callback)

分片上传本地文件

aliyun_client.multipart_upload_file(key='xxx.pth', file='xxxx.pth')

如果需要知道上传进度,可以:

def progress_callback(block_size, total_size):
    print(f"正在上传: {block_size / total_size * 100:.2f}%")

aliyun_client.multipart_upload_file(key='xxx.pth', file='xxx.pth', progress_callback=progress_callback)