deep-high-resolution-net.pytorch

最新推荐文章于 2024-08-19 23:23:47 发布

AI算法网奇

最新推荐文章于 2024-08-19 23:23:47 发布

阅读量3.1k

点赞数

分类专栏：深度学习宝典

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

版权

深度学习宝典专栏收录该内容

998 篇文章 721 订阅 ¥29.90 ¥99.00

订阅专栏

超级会员免费看

本文介绍了一个名为'deep-high-resolution-net.pytorch'的项目，该实现聚焦于深度学习中的高分辨率网络，在PyTorch框架下运行，平均耗时约100多毫秒。

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

deep-high-resolution-net.pytorch

1070 100多ms


from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import argparse
import os
import pprint

import cv2
import torch
import torch.nn.parallel
import torch.backends.cudnn as cudnn
import torch.optim
import torch.utils.data
import torch.utils.data.distributed
import torchvision.transforms as transforms
from easydict import EasyDict as edict
import _init_paths
from config import cfg
from config import update_config
from core.loss import JointsMSELoss
from core.function import validate
from utils.utils import create_logger

from utils.transforms import flip_back
import dataset
import models


def parse_args():

了解本专栏