torch,权重400多m
https://github.com/JXingZhao/EGNet
这个是画掩码的,不是目标检测,是目标显著性检测,出来是二值图
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import argparse
import os
import time
import cv2
import torch
import numpy as np
from dataset import get_loader
from model import TUN_bone, extra_layer
from solver import Solver
from vgg import vgg16
def main(config):
if config.mode == 'train':
train_loader, dataset = get_loader(config.batch_size, num_thread=config.num_thread)
run = "nnet"
if not os.path.exists("%s/run-%s" % (config.save_fold, run)):
os.mkdir("%s/run-%s" %