Python深度学习代码块

监控显存

watch -n 1 nvidia-smi

图像分类任务图片裁剪

ImageNet数据集预处理：如果一个图片的长宽比例小于一定比值，则直接拉伸。如果大于一定比值，按照 左、中、右 三部分进行切割

def process(img,max_ratio=1.35):
    if len(img.shape)==2:
        img=cv2.cvtColor(img,cv2.COLOR_GRAY2RGB)
    h,w,_=img.shape
    ratio=h/w if h>w else w/h
    print(ratio)
    if ratio<max_ratio:
        return [img]
        # return [cv2.resize(img,size)]
    isRotated=False
    if h>w:
        isRotated