由于卷积方式的改变需要对模型结构进行一些修改,因此这里给出完整的修改代码。

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

import os
import logging

import torch
import torch.nn as nn
import torch.nn.functional as F


BatchNorm2d = nn.BatchNorm2d
BN_MOMENTUM = 0.01
logger
请将这部分代码中卷积方式替换为深度可分离卷积# ------------------------------------------------------------------------------# Copyright c Microsoft# Licensed under the MIT License# Create by Bin Xiao BinXiaomicrosoftcom# M

原文地址: https://www.cveoy.top/t/topic/faCf 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录