当前位置:网站首页>paddlepaddle 21 基于dropout实现用4行代码dropblock

paddlepaddle 21 基于dropout实现用4行代码dropblock

2022-06-27 01:58:00 万里鹏程转瞬至

dropblock的作用不在累述,这里只分享使用4行代码实现dropblock。dropblock的介绍可以参考博主的另一篇文章pytorch 22 8种Dropout方法的简介 及 基于Dropout用4行代码快速实现DropBlock_万里鹏程转瞬至的博客-CSDN博客

1、基本库方法实现

这里主要实现三个函数,图片展示、图片读取和tensor转img

from PIL import Image
from matplotlib import pyplot as plt
import paddle
from paddle import nn
def myimshows(imgs, titles=False, fname="test.jpg", size=6):
    lens = len(imgs)
    fig = plt.figure(figsize=(size * lens,size))
    if titles == False:
        titles="0123456789"
    for i in range(1, lens + 1):
        cols = 100 + lens * 10 + i
        plt.xticks(())
        plt.yticks(())
        plt.subplot(cols)
        if len(imgs[i - 1].shape) == 2:
            plt.imshow(imgs[i - 1], cmap='Reds'
原网站

版权声明
本文为[万里鹏程转瞬至]所创,转载请带上原文链接,感谢
https://hpg123.blog.csdn.net/article/details/123964892