基于Android pytorch API 深度学习语义分割（下）

草稿

import numpy as np
import torch.nn.functional as F
import cv2
from dataset import Mydataset
from fcNet import *
from torch.utils.data import DataLoader
import torch.nn as nn
import torch
from eval import *

numclass = 2
epoches = 100
train_data = Mydataset(352, 480, True)
train_loder = DataLoader(train_data, batch_size=8, shuffle=True, num_workers=0)
val_data = Mydataset(352, 480, False)
val_loder = DataLoader(train_data, batch_size=8, shuffle=True, num_workers=0)
fcn=MyFcNet_vgg16bn(2)
cretion = nn.NLLLoss()
optimizer = torch.optim.Adam(fcn.parameters(), lr=1e-4)



def train(model):
    best=[0]
    net = model.train()

    for epoch in range(epoches):
        print("Epoch is [{}/{}]".format(epoch + 1, epoches))
        if epoch % 50 == 0 and epoch != 0:
            for gp in optimizer.param_groups:
                gp['lr'] *= 0.5

        train_loss =