强行分类提取特征自编码网络例3

# -*- coding: utf-8 -*-
import torch
import cv2
import numpy as np
import matplotlib.pyplot as plt
from PIL import Image, ImageDraw, ImageFont
class S_net(torch.nn.Module):
    def __init__