Add CenterNet based on Resnet101, TensorRT not implemented.
Signed-off-by: Davide Sapienza <sapienza.dav@gmail.com>
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user