模型43M,1060显卡 640*400 图像耗时1-2毫秒,速度挺快的。
开源地址:
demo效果:
自己编写了读取图片可视化的脚本代码:
import os
import sys
import time
import cv2
import numpy as np
import torch
import argparse
import torch.utils
import torch.backends.cudnn as cudnn
from PIL import Image
from torch.autograd import Variable
from model import Finetunemodel
from multi_read_data import MemoryFriendlyLoader
parser = argparse.ArgumentParser("