ceccocats / tkDNN

Deep neural network library and toolkit to do high performace inference on NVIDIA jetson platforms
GNU General Public License v2.0
718 stars 209 forks source link

Channels = 4 #211

Closed Hiroaki-K4 closed 3 years ago

Hiroaki-K4 commented 3 years ago

Hi. I customized yolo and trained it on 4 channels. There are 4 channels, RGB and depth. Is it possible to export this 4-channel weights file?

[net]
# Testing
#batch=1
#subdivisions=1
# Training
batch=1
subdivisions=1
width=512
height=512
channels=4
momentum=0.949
decay=0.0005
angle=0
saturation = 1.5
exposure = 1.5
hue=.1

learning_rate=0.001
burn_in=1000
max_batches = 500500
policy=steps
steps=400000,450000
scales=.1,.1

The config file used for training is as follows. yolov4.txt