mobilenet v4 导出onnx onnx推理

目录

torch推理：

导出onnx：

onnx推理：

---

torch推理：

# -*- coding: utf-8 -*-
from __future__ import print_function, division
import argparse
import configparser
import shutil
from collections import OrderedDict

import cv2
import numpy as np
import torch

import torch.nn.functional as torch_F

from mobilenet.mobilenetv4 import MobileNetV4

import time
import os

use_gpu = torch.cuda.is_available()

class TimeStr:
    def __init__(self):
        self.last_time, self.counter = '', 0

    def get_time(self, fmt="%m%d_%H%M_%S"):
        now_time = time.strftime(fmt)
        self.counter += 1
        if now_time != self.last_time:
            self.last_time