1070 416/512 1 9ms
1070 416/512 16 90ms
Android版:
代码参考:https://blog.csdn.net/shanglianlm/article/details/95809339
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import time
import torch
import torch.nn as nn
class BlazeBlock(nn.Module):
def __init__(self, in_channels,out_channels,mid_channels=None,stride=1):
super(BlazeBlock, self).__init__()
mid_chan