EGNet

最新推荐文章于 2022-03-26 08:38:19 发布

AI算法网奇

最新推荐文章于 2022-03-26 08:38:19 发布

阅读量3.3k

点赞数

分类专栏： pytorch知识宝典

本文链接：https://blog.csdn.net/jacke121/article/details/100165601

版权

pytorch知识宝典专栏收录该内容

499 篇文章 226 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

EGNet是一个用于目标显著性检测的模型，其权重文件大小超过400MB，基于torch实现。该项目在GitHub上开源，产出结果为二值图，主要关注目标的掩码绘制，而非目标检测。

摘要由CSDN通过智能技术生成

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" %

了解本专栏