模型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("