笔记
demo12138
这个作者很懒,什么都没留下…
展开
-
安检数据集备份
链接:https://pan.baidu.com/s/1p95nzPy0LzxfjYixICzeiA 提取码:sofm原创 2020-09-23 15:20:31 · 762 阅读 · 4 评论 -
2020-09-21
from future import divisionimport torchimport torch.nn as nnimport torch.nn.functional as Ffrom torch.autograd import Variableimport numpy as npfrom net.fcn_parts import *class FCNet1(nn.Module):def init(self, channels, classes, drop_rate):super()原创 2020-09-21 19:09:00 · 121 阅读 · 0 评论 -
2020-09-21
import osimport torch as timport torch.nn as nnfrom torchvision import transformsimport cv2from torchvision.transforms.functional import to_tensorfrom unet import Unetimport numpy as npimport torch.nn.functional as Ffrom fcn import FCNet1def pred原创 2020-09-21 18:40:32 · 117 阅读 · 0 评论 -
2020-09-21
from torch.autograd import Variableimport torch as tfrom torch.utils.data import DataLoaderfrom prepareData import MyDatasetfrom torch import nn, optimfrom unet import Unetimport osfrom fcn import FCNet1device = t.device(“cuda” if t.cuda.is_availab原创 2020-09-21 17:10:34 · 110 阅读 · 0 评论 -
2020-09-21
import argparseimport loggingimport osimport sysimport numpy as npimport torchimport torch.nn as nnfrom torch import optimfrom tqdm import tqdmfrom eval import eval_netfrom unet import UNetfrom unet.fcn1 import *from torch.utils.tensorboard imp原创 2020-09-21 16:32:25 · 315 阅读 · 0 评论