NVIDIA-AI-IOT / redtail

Perception and AI components for autonomous mobile robotics.
BSD 3-Clause "New" or "Revised" License
1.01k stars 346 forks source link

Training Resnet18_2D #149

Open dinimar opened 4 years ago

dinimar commented 4 years ago

I want to train ResNet18_2D model on custom dataset (indoor environment) cause model trained on KITTI dataset works badly on room images. So, could you provide tensorflow code with implementation of this model to make custom checkpoint file.