Add CenterNet based on Resnet101, TensorRT not implemented.

Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
Davide Sapienza
2019-12-18 18:59:54 +01:00
parent 44b2bce3ff
commit df888a3457
13 changed files with 773 additions and 11 deletions
@@ -6,6 +6,8 @@ from torchsummary import summary
import numpy as np
import struct
import torch.nn as nn
def bin_write(f, data):
data =data.flatten()
# print(data)