├── .gitignore ├── LICENSE ├── README.md ├── pytorch ├── action_recognition_net │ ├── AppOFCuda │ ├── actionrecognitionnet.ipynb │ └── specs │ │ ├── evaluate_of.yaml │ │ ├── evaluate_rgb.yaml │ │ ├── export_of.yaml │ │ ├── export_rgb.yaml │ │ ├── infer_of.yaml │ │ ├── infer_rgb.yaml │ │ ├── train_of_3d_finetune.yaml │ │ ├── train_rgb_2d_finetune.yaml │ │ └── train_rgb_3d_finetune.yaml ├── pose_classification_net │ ├── poseclassificationnet.ipynb │ └── specs │ │ ├── experiment_kinetics.yaml │ │ └── experiment_nvidia.yaml ├── requirements-pip-pytorch.txt ├── requirements-pip.txt ├── setup_env.sh └── setup_env_desktop.sh ├── readme_images ├── colab_view_files_1.png ├── colab_view_files_2.png └── colab_view_files_3.png ├── tao_deploy ├── setup_env_colab.sh ├── specs │ ├── DashCamNet │ │ ├── DashCamNet_infer.txt │ │ └── DashCamNet_trt.txt │ ├── FaceDetect │ │ ├── FaceDetect_infer.txt │ │ └── FaceDetect_trt.txt │ ├── FaceDetectIR │ │ ├── FaceDetectIR_infer.txt │ │ └── FaceDetectIR_trt.txt │ ├── LicensePlateDetection │ │ ├── LicensePlateDetection_infer.txt │ │ └── LicensePlateDetection_trt.txt │ ├── LicensePlateRecognition │ │ └── LicensePlateRecognition.txt │ ├── PeopleNet │ │ ├── PeopleNet_infer.txt │ │ └── PeopleNet_trt.txt │ ├── PeopleSemSegNet │ │ ├── PeopleSemSegNet.txt │ │ └── PeopleSemSegNet_data.txt │ ├── TrafficCamNet │ │ ├── TrafficCamNet_infer.txt │ │ └── TrafficCamNet_trt.txt │ ├── VehicleMakeNet │ │ ├── VehicleMakeNet.txt │ │ └── classmap.json │ └── VehicleTypeNet │ │ ├── VehicleTypeNet.txt │ │ └── classmap.json └── tao_deploy.ipynb └── tensorflow ├── classification ├── classification.ipynb └── specs │ ├── classification_retrain_spec.cfg │ └── classification_spec.cfg ├── dssd ├── __init__.py ├── dssd.ipynb ├── generate_val_dataset.py └── specs │ ├── dssd_retrain_resnet18_kitti.txt │ ├── dssd_tfrecords_kitti_train.txt │ └── dssd_train_resnet18_kitti.txt ├── efficientdet ├── efficientdet.ipynb └── specs │ ├── coco_labels.txt │ ├── create_coco_tf_record.py │ ├── download_coco.sh │ ├── efficientdet_d0_retrain.txt │ └── efficientdet_d0_train.txt ├── lprnet ├── download_and_prepare_data.sh ├── lprnet.ipynb ├── preprocess_openalpr_benchmark.py └── specs │ ├── tutorial_spec.txt │ └── us_lp_characters.txt ├── mask_rcnn ├── maskrcnn.ipynb └── specs │ ├── coco_labels.txt │ ├── create_coco_tf_record.py │ ├── download_coco.sh │ ├── maskrcnn_retrain_resnet50.txt │ └── maskrcnn_train_resnet50.txt ├── multitask_classification ├── __init__.py ├── multitask_classification.ipynb └── specs │ ├── mclassification_retrain_spec.cfg │ └── mclassification_spec.cfg ├── requirements-pip-desktop.txt ├── requirements-pip.txt ├── retinanet ├── __init__.py ├── generate_val_dataset.py ├── retinanet.ipynb └── specs │ ├── retinanet_retrain_resnet18_kitti.txt │ ├── retinanet_tfrecords_kitti_train.txt │ └── retinanet_train_resnet18_kitti.txt ├── setup_env.sh ├── setup_env_desktop.sh ├── ssd ├── __init__.py ├── generate_val_dataset.py ├── specs │ ├── ssd_retrain_resnet18_kitti.txt │ ├── ssd_tfrecords_kitti_train.txt │ └── ssd_train_resnet18_kitti.txt └── ssd.ipynb ├── unet ├── prepare_data.sh ├── prepare_data_isbi.py ├── specs │ ├── unet_train_resnet_unet_isbi.txt │ └── unet_train_resnet_unet_isbi_retrain.txt ├── unet_isbi.ipynb └── vis_annotation.py ├── utils ├── generate_coco_subset.py └── generate_kitti_subset.py ├── yolo_v3 ├── __init__.py ├── specs │ ├── tfrecords_kitti_trainval.txt │ ├── yolo_v3_retrain_resnet18_tfrecord.txt │ └── yolo_v3_train_resnet18_tfrecord.txt └── yolo_v3.ipynb ├── yolo_v4 ├── __init__.py ├── specs │ ├── yolo_v4_retrain_resnet18_kitti.txt │ ├── yolo_v4_tfrecords_kitti_train.txt │ ├── yolo_v4_tfrecords_kitti_val.txt │ └── yolo_v4_train_resnet18_kitti.txt └── yolo_v4.ipynb └── yolo_v4_tiny ├── __init__.py ├── specs ├── yolo_v4_tiny_retrain_kitti.txt ├── yolo_v4_tiny_tfrecords_kitti_train.txt ├── yolo_v4_tiny_tfrecords_kitti_val.txt └── yolo_v4_tiny_train_kitti.txt └── yolo_v4_tiny.ipynb /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/README.md -------------------------------------------------------------------------------- /pytorch/action_recognition_net/AppOFCuda: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/AppOFCuda -------------------------------------------------------------------------------- /pytorch/action_recognition_net/actionrecognitionnet.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/actionrecognitionnet.ipynb -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/evaluate_of.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/evaluate_of.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/evaluate_rgb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/evaluate_rgb.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/export_of.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/export_of.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/export_rgb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/export_rgb.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/infer_of.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/infer_of.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/infer_rgb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/infer_rgb.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/train_of_3d_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/train_of_3d_finetune.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/train_rgb_2d_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/train_rgb_2d_finetune.yaml -------------------------------------------------------------------------------- /pytorch/action_recognition_net/specs/train_rgb_3d_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/action_recognition_net/specs/train_rgb_3d_finetune.yaml -------------------------------------------------------------------------------- /pytorch/pose_classification_net/poseclassificationnet.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/pose_classification_net/poseclassificationnet.ipynb -------------------------------------------------------------------------------- /pytorch/pose_classification_net/specs/experiment_kinetics.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/pose_classification_net/specs/experiment_kinetics.yaml -------------------------------------------------------------------------------- /pytorch/pose_classification_net/specs/experiment_nvidia.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/pose_classification_net/specs/experiment_nvidia.yaml -------------------------------------------------------------------------------- /pytorch/requirements-pip-pytorch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/requirements-pip-pytorch.txt -------------------------------------------------------------------------------- /pytorch/requirements-pip.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/requirements-pip.txt -------------------------------------------------------------------------------- /pytorch/setup_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/setup_env.sh -------------------------------------------------------------------------------- /pytorch/setup_env_desktop.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/pytorch/setup_env_desktop.sh -------------------------------------------------------------------------------- /readme_images/colab_view_files_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/readme_images/colab_view_files_1.png -------------------------------------------------------------------------------- /readme_images/colab_view_files_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/readme_images/colab_view_files_2.png -------------------------------------------------------------------------------- /readme_images/colab_view_files_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/readme_images/colab_view_files_3.png -------------------------------------------------------------------------------- /tao_deploy/setup_env_colab.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/setup_env_colab.sh -------------------------------------------------------------------------------- /tao_deploy/specs/DashCamNet/DashCamNet_infer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/DashCamNet/DashCamNet_infer.txt -------------------------------------------------------------------------------- /tao_deploy/specs/DashCamNet/DashCamNet_trt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/DashCamNet/DashCamNet_trt.txt -------------------------------------------------------------------------------- /tao_deploy/specs/FaceDetect/FaceDetect_infer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/FaceDetect/FaceDetect_infer.txt -------------------------------------------------------------------------------- /tao_deploy/specs/FaceDetect/FaceDetect_trt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/FaceDetect/FaceDetect_trt.txt -------------------------------------------------------------------------------- /tao_deploy/specs/FaceDetectIR/FaceDetectIR_infer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/FaceDetectIR/FaceDetectIR_infer.txt -------------------------------------------------------------------------------- /tao_deploy/specs/FaceDetectIR/FaceDetectIR_trt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/FaceDetectIR/FaceDetectIR_trt.txt -------------------------------------------------------------------------------- /tao_deploy/specs/LicensePlateDetection/LicensePlateDetection_infer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/LicensePlateDetection/LicensePlateDetection_infer.txt -------------------------------------------------------------------------------- /tao_deploy/specs/LicensePlateDetection/LicensePlateDetection_trt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/LicensePlateDetection/LicensePlateDetection_trt.txt -------------------------------------------------------------------------------- /tao_deploy/specs/LicensePlateRecognition/LicensePlateRecognition.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/LicensePlateRecognition/LicensePlateRecognition.txt -------------------------------------------------------------------------------- /tao_deploy/specs/PeopleNet/PeopleNet_infer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/PeopleNet/PeopleNet_infer.txt -------------------------------------------------------------------------------- /tao_deploy/specs/PeopleNet/PeopleNet_trt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/PeopleNet/PeopleNet_trt.txt -------------------------------------------------------------------------------- /tao_deploy/specs/PeopleSemSegNet/PeopleSemSegNet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/PeopleSemSegNet/PeopleSemSegNet.txt -------------------------------------------------------------------------------- /tao_deploy/specs/PeopleSemSegNet/PeopleSemSegNet_data.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/PeopleSemSegNet/PeopleSemSegNet_data.txt -------------------------------------------------------------------------------- /tao_deploy/specs/TrafficCamNet/TrafficCamNet_infer.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/TrafficCamNet/TrafficCamNet_infer.txt -------------------------------------------------------------------------------- /tao_deploy/specs/TrafficCamNet/TrafficCamNet_trt.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/TrafficCamNet/TrafficCamNet_trt.txt -------------------------------------------------------------------------------- /tao_deploy/specs/VehicleMakeNet/VehicleMakeNet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/VehicleMakeNet/VehicleMakeNet.txt -------------------------------------------------------------------------------- /tao_deploy/specs/VehicleMakeNet/classmap.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/VehicleMakeNet/classmap.json -------------------------------------------------------------------------------- /tao_deploy/specs/VehicleTypeNet/VehicleTypeNet.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/VehicleTypeNet/VehicleTypeNet.txt -------------------------------------------------------------------------------- /tao_deploy/specs/VehicleTypeNet/classmap.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/specs/VehicleTypeNet/classmap.json -------------------------------------------------------------------------------- /tao_deploy/tao_deploy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tao_deploy/tao_deploy.ipynb -------------------------------------------------------------------------------- /tensorflow/classification/classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/classification/classification.ipynb -------------------------------------------------------------------------------- /tensorflow/classification/specs/classification_retrain_spec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/classification/specs/classification_retrain_spec.cfg -------------------------------------------------------------------------------- /tensorflow/classification/specs/classification_spec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/classification/specs/classification_spec.cfg -------------------------------------------------------------------------------- /tensorflow/dssd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/dssd/__init__.py -------------------------------------------------------------------------------- /tensorflow/dssd/dssd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/dssd/dssd.ipynb -------------------------------------------------------------------------------- /tensorflow/dssd/generate_val_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/dssd/generate_val_dataset.py -------------------------------------------------------------------------------- /tensorflow/dssd/specs/dssd_retrain_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/dssd/specs/dssd_retrain_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/dssd/specs/dssd_tfrecords_kitti_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/dssd/specs/dssd_tfrecords_kitti_train.txt -------------------------------------------------------------------------------- /tensorflow/dssd/specs/dssd_train_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/dssd/specs/dssd_train_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/efficientdet/efficientdet.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/efficientdet/efficientdet.ipynb -------------------------------------------------------------------------------- /tensorflow/efficientdet/specs/coco_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/efficientdet/specs/coco_labels.txt -------------------------------------------------------------------------------- /tensorflow/efficientdet/specs/create_coco_tf_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/efficientdet/specs/create_coco_tf_record.py -------------------------------------------------------------------------------- /tensorflow/efficientdet/specs/download_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/efficientdet/specs/download_coco.sh -------------------------------------------------------------------------------- /tensorflow/efficientdet/specs/efficientdet_d0_retrain.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/efficientdet/specs/efficientdet_d0_retrain.txt -------------------------------------------------------------------------------- /tensorflow/efficientdet/specs/efficientdet_d0_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/efficientdet/specs/efficientdet_d0_train.txt -------------------------------------------------------------------------------- /tensorflow/lprnet/download_and_prepare_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/lprnet/download_and_prepare_data.sh -------------------------------------------------------------------------------- /tensorflow/lprnet/lprnet.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/lprnet/lprnet.ipynb -------------------------------------------------------------------------------- /tensorflow/lprnet/preprocess_openalpr_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/lprnet/preprocess_openalpr_benchmark.py -------------------------------------------------------------------------------- /tensorflow/lprnet/specs/tutorial_spec.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/lprnet/specs/tutorial_spec.txt -------------------------------------------------------------------------------- /tensorflow/lprnet/specs/us_lp_characters.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/lprnet/specs/us_lp_characters.txt -------------------------------------------------------------------------------- /tensorflow/mask_rcnn/maskrcnn.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/mask_rcnn/maskrcnn.ipynb -------------------------------------------------------------------------------- /tensorflow/mask_rcnn/specs/coco_labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/mask_rcnn/specs/coco_labels.txt -------------------------------------------------------------------------------- /tensorflow/mask_rcnn/specs/create_coco_tf_record.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/mask_rcnn/specs/create_coco_tf_record.py -------------------------------------------------------------------------------- /tensorflow/mask_rcnn/specs/download_coco.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/mask_rcnn/specs/download_coco.sh -------------------------------------------------------------------------------- /tensorflow/mask_rcnn/specs/maskrcnn_retrain_resnet50.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/mask_rcnn/specs/maskrcnn_retrain_resnet50.txt -------------------------------------------------------------------------------- /tensorflow/mask_rcnn/specs/maskrcnn_train_resnet50.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/mask_rcnn/specs/maskrcnn_train_resnet50.txt -------------------------------------------------------------------------------- /tensorflow/multitask_classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/multitask_classification/__init__.py -------------------------------------------------------------------------------- /tensorflow/multitask_classification/multitask_classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/multitask_classification/multitask_classification.ipynb -------------------------------------------------------------------------------- /tensorflow/multitask_classification/specs/mclassification_retrain_spec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/multitask_classification/specs/mclassification_retrain_spec.cfg -------------------------------------------------------------------------------- /tensorflow/multitask_classification/specs/mclassification_spec.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/multitask_classification/specs/mclassification_spec.cfg -------------------------------------------------------------------------------- /tensorflow/requirements-pip-desktop.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | tqdm 3 | opencv-python 4 | h5py 5 | pyyaml 6 | pandas 7 | Pillow 8 | -------------------------------------------------------------------------------- /tensorflow/requirements-pip.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/requirements-pip.txt -------------------------------------------------------------------------------- /tensorflow/retinanet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/retinanet/__init__.py -------------------------------------------------------------------------------- /tensorflow/retinanet/generate_val_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/retinanet/generate_val_dataset.py -------------------------------------------------------------------------------- /tensorflow/retinanet/retinanet.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/retinanet/retinanet.ipynb -------------------------------------------------------------------------------- /tensorflow/retinanet/specs/retinanet_retrain_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/retinanet/specs/retinanet_retrain_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/retinanet/specs/retinanet_tfrecords_kitti_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/retinanet/specs/retinanet_tfrecords_kitti_train.txt -------------------------------------------------------------------------------- /tensorflow/retinanet/specs/retinanet_train_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/retinanet/specs/retinanet_train_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/setup_env.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/setup_env.sh -------------------------------------------------------------------------------- /tensorflow/setup_env_desktop.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/setup_env_desktop.sh -------------------------------------------------------------------------------- /tensorflow/ssd/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/ssd/__init__.py -------------------------------------------------------------------------------- /tensorflow/ssd/generate_val_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/ssd/generate_val_dataset.py -------------------------------------------------------------------------------- /tensorflow/ssd/specs/ssd_retrain_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/ssd/specs/ssd_retrain_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/ssd/specs/ssd_tfrecords_kitti_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/ssd/specs/ssd_tfrecords_kitti_train.txt -------------------------------------------------------------------------------- /tensorflow/ssd/specs/ssd_train_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/ssd/specs/ssd_train_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/ssd/ssd.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/ssd/ssd.ipynb -------------------------------------------------------------------------------- /tensorflow/unet/prepare_data.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/unet/prepare_data.sh -------------------------------------------------------------------------------- /tensorflow/unet/prepare_data_isbi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/unet/prepare_data_isbi.py -------------------------------------------------------------------------------- /tensorflow/unet/specs/unet_train_resnet_unet_isbi.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/unet/specs/unet_train_resnet_unet_isbi.txt -------------------------------------------------------------------------------- /tensorflow/unet/specs/unet_train_resnet_unet_isbi_retrain.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/unet/specs/unet_train_resnet_unet_isbi_retrain.txt -------------------------------------------------------------------------------- /tensorflow/unet/unet_isbi.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/unet/unet_isbi.ipynb -------------------------------------------------------------------------------- /tensorflow/unet/vis_annotation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/unet/vis_annotation.py -------------------------------------------------------------------------------- /tensorflow/utils/generate_coco_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/utils/generate_coco_subset.py -------------------------------------------------------------------------------- /tensorflow/utils/generate_kitti_subset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/utils/generate_kitti_subset.py -------------------------------------------------------------------------------- /tensorflow/yolo_v3/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v3/__init__.py -------------------------------------------------------------------------------- /tensorflow/yolo_v3/specs/tfrecords_kitti_trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v3/specs/tfrecords_kitti_trainval.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v3/specs/yolo_v3_retrain_resnet18_tfrecord.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v3/specs/yolo_v3_retrain_resnet18_tfrecord.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v3/specs/yolo_v3_train_resnet18_tfrecord.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v3/specs/yolo_v3_train_resnet18_tfrecord.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v3/yolo_v3.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v3/yolo_v3.ipynb -------------------------------------------------------------------------------- /tensorflow/yolo_v4/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4/__init__.py -------------------------------------------------------------------------------- /tensorflow/yolo_v4/specs/yolo_v4_retrain_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4/specs/yolo_v4_retrain_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4/specs/yolo_v4_tfrecords_kitti_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4/specs/yolo_v4_tfrecords_kitti_train.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4/specs/yolo_v4_tfrecords_kitti_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4/specs/yolo_v4_tfrecords_kitti_val.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4/specs/yolo_v4_train_resnet18_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4/specs/yolo_v4_train_resnet18_kitti.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4/yolo_v4.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4/yolo_v4.ipynb -------------------------------------------------------------------------------- /tensorflow/yolo_v4_tiny/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4_tiny/__init__.py -------------------------------------------------------------------------------- /tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_retrain_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_retrain_kitti.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_kitti_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_kitti_train.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_kitti_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_tfrecords_kitti_val.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_train_kitti.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4_tiny/specs/yolo_v4_tiny_train_kitti.txt -------------------------------------------------------------------------------- /tensorflow/yolo_v4_tiny/yolo_v4_tiny.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NVIDIA-AI-IOT/nvidia-tao/HEAD/tensorflow/yolo_v4_tiny/yolo_v4_tiny.ipynb --------------------------------------------------------------------------------