gitextract_1s0rnibl/ ├── .github/ │ └── ISSUE_TEMPLATE/ │ └── issue-template.md ├── .gitignore ├── LICENSE ├── README.md ├── article_link/ │ └── README.md ├── course_ppt/ │ └── README.md ├── data_set/ │ ├── README.md │ └── split_data.py ├── deploying_service/ │ ├── deploying_pytorch/ │ │ ├── convert_onnx_cls/ │ │ │ ├── class_indices.json │ │ │ ├── main.py │ │ │ └── model.py │ │ ├── convert_openvino/ │ │ │ ├── convert_resnet34/ │ │ │ │ ├── README.md │ │ │ │ ├── compare_fps.py │ │ │ │ ├── compare_onnx_and_ir.py │ │ │ │ ├── convert_pytorch2onnx.py │ │ │ │ ├── model.py │ │ │ │ ├── quantization_int8.py │ │ │ │ ├── requirements.txt │ │ │ │ └── utils.py │ │ │ └── convert_yolov5/ │ │ │ ├── README.md │ │ │ ├── compare_fps.py │ │ │ ├── compare_onnx_and_ir.py │ │ │ ├── draw_box_utils.py │ │ │ ├── evaluation.py │ │ │ ├── predict.py │ │ │ ├── quantization_int8.py │ │ │ ├── requirements.txt │ │ │ └── utils.py │ │ ├── convert_tensorrt/ │ │ │ └── convert_resnet34/ │ │ │ ├── compare_onnx_and_trt.py │ │ │ ├── convert_pytorch2onnx.py │ │ │ ├── my_dataset.py │ │ │ ├── quantization.py │ │ │ └── utils.py │ │ └── pytorch_flask_service/ │ │ ├── class_indices.json │ │ ├── main.py │ │ ├── model.py │ │ ├── requirements.txt │ │ └── templates/ │ │ └── up.html │ └── pruning_model_pytorch/ │ ├── class_indices.json │ ├── main.py │ ├── model.py │ ├── predict.py │ └── train.py ├── others_project/ │ ├── draw_dilated_conv/ │ │ └── main.py │ ├── kmeans_anchors/ │ │ ├── main.py │ │ ├── plot_kmeans.py │ │ ├── read_voc.py │ │ └── yolo_kmeans.py │ ├── openvinotest/ │ │ └── openvino_cls_test/ │ │ ├── class_indices.json │ │ ├── create_imagenet_annotation.py │ │ ├── float32vsint8.py │ │ ├── main.py │ │ ├── model.py │ │ └── speed_test.py │ ├── readPbFile/ │ │ ├── README.md │ │ ├── pascal_label_map.pbtxt │ │ ├── readPb.py │ │ ├── test_images/ │ │ │ └── image_info.txt │ │ └── using_function.py │ ├── textcnnKeras/ │ │ ├── dataGenerator.py │ │ ├── data_link.txt │ │ ├── main.py │ │ └── models.py │ └── trans_widerface_to_xml/ │ ├── create_xml.py │ └── main.py ├── pytorch_classification/ │ ├── ConfusionMatrix/ │ │ ├── class_indices.json │ │ ├── main.py │ │ └── model.py │ ├── ConvNeXt/ │ │ ├── README.md │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── train.py │ │ └── utils.py │ ├── MobileViT/ │ │ ├── README.md │ │ ├── model.py │ │ ├── model_config.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── train.py │ │ ├── transformer.py │ │ ├── unfold_test.py │ │ └── utils.py │ ├── README.md │ ├── Test10_regnet/ │ │ ├── README.md │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── pretrain_weights.py │ │ ├── train.py │ │ └── utils.py │ ├── Test11_efficientnetV2/ │ │ ├── README.md │ │ ├── class_indices.json │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── train.py │ │ ├── trans_effv2_weights.py │ │ └── utils.py │ ├── Test1_official_demo/ │ │ ├── model.py │ │ ├── predict.py │ │ └── train.py │ ├── Test2_alexnet/ │ │ ├── class_indices.json │ │ ├── model.py │ │ ├── predict.py │ │ └── train.py │ ├── Test3_vggnet/ │ │ ├── class_indices.json │ │ ├── model.py │ │ ├── predict.py │ │ └── train.py │ ├── Test4_googlenet/ │ │ ├── class_indices.json │ │ ├── model.py │ │ ├── predict.py │ │ └── train.py │ ├── Test5_resnet/ │ │ ├── README.md │ │ ├── batch_predict.py │ │ ├── class_indices.json │ │ ├── load_weights.py │ │ ├── model.py │ │ ├── predict.py │ │ └── train.py │ ├── Test6_mobilenet/ │ │ ├── class_indices.json │ │ ├── model_v2.py │ │ ├── model_v3.py │ │ ├── predict.py │ │ └── train.py │ ├── Test7_shufflenet/ │ │ ├── README.md │ │ ├── class_indices.json │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── train.py │ │ └── utils.py │ ├── Test8_densenet/ │ │ ├── README.md │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── train.py │ │ └── utils.py │ ├── Test9_efficientNet/ │ │ ├── README.md │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── train.py │ │ ├── trans_weights_to_pytorch.py │ │ └── utils.py │ ├── analyze_weights_featuremap/ │ │ ├── alexnet_model.py │ │ ├── analyze_feature_map.py │ │ ├── analyze_kernel_weight.py │ │ └── resnet_model.py │ ├── custom_dataset/ │ │ ├── main.py │ │ ├── my_dataset.py │ │ └── utils.py │ ├── grad_cam/ │ │ ├── README.md │ │ ├── imagenet1k_classes.txt │ │ ├── imagenet21k_classes.txt │ │ ├── main_cnn.py │ │ ├── main_swin.py │ │ ├── main_vit.py │ │ ├── swin_model.py │ │ ├── utils.py │ │ └── vit_model.py │ ├── mini_imagenet/ │ │ ├── README.md │ │ ├── imagenet_class_index.json │ │ ├── model.py │ │ ├── multi_train_utils/ │ │ │ ├── __init__.py │ │ │ ├── distributed_utils.py │ │ │ └── train_eval_utils.py │ │ ├── my_dataset.py │ │ ├── restructure_csv.py │ │ ├── train_multi_gpu_using_launch.py │ │ └── train_single_gpu.py │ ├── model_complexity/ │ │ ├── main.py │ │ ├── model.py │ │ └── utils.py │ ├── swin_transformer/ │ │ ├── README.md │ │ ├── create_confusion_matrix.py │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── select_incorrect_samples.py │ │ ├── train.py │ │ └── utils.py │ ├── tensorboard_test/ │ │ ├── data_utils.py │ │ ├── model.py │ │ ├── my_dataset.py │ │ ├── requirements.txt │ │ ├── train.py │ │ └── train_eval_utils.py │ ├── train_multi_GPU/ │ │ ├── README.md │ │ ├── model.py │ │ ├── multi_train_utils/ │ │ │ ├── distributed_utils.py │ │ │ └── train_eval_utils.py │ │ ├── my_dataset.py │ │ ├── plot_results.py │ │ ├── requirements.txt │ │ ├── train_multi_gpu_using_launch.py │ │ ├── train_multi_gpu_using_spawn.py │ │ ├── train_single_gpu.py │ │ └── utils.py │ └── vision_transformer/ │ ├── README.md │ ├── flops.py │ ├── my_dataset.py │ ├── predict.py │ ├── train.py │ ├── utils.py │ └── vit_model.py ├── pytorch_keypoint/ │ ├── DeepPose/ │ │ ├── README.md │ │ ├── datasets.py │ │ ├── export_onnx.py │ │ ├── model.py │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── distributed_utils.py │ │ │ ├── losses.py │ │ │ ├── metrics.py │ │ │ └── train_eval_utils.py │ │ ├── transforms.py │ │ ├── utils.py │ │ └── wflw_horizontal_flip_indices.py │ └── HRNet/ │ ├── README.md │ ├── draw_utils.py │ ├── model/ │ │ ├── __init__.py │ │ └── hrnet.py │ ├── my_dataset_coco.py │ ├── person_keypoints.json │ ├── plot_curve.py │ ├── predict.py │ ├── requirements.txt │ ├── train.py │ ├── train_multi_GPU.py │ ├── train_utils/ │ │ ├── __init__.py │ │ ├── coco_eval.py │ │ ├── coco_utils.py │ │ ├── distributed_utils.py │ │ ├── group_by_aspect_ratio.py │ │ ├── loss.py │ │ └── train_eval_utils.py │ ├── transforms.py │ └── validation.py ├── pytorch_object_detection/ │ ├── faster_rcnn/ │ │ ├── README.md │ │ ├── backbone/ │ │ │ ├── __init__.py │ │ │ ├── feature_pyramid_network.py │ │ │ ├── mobilenetv2_model.py │ │ │ ├── resnet50_fpn_model.py │ │ │ └── vgg_model.py │ │ ├── change_backbone_with_fpn.py │ │ ├── change_backbone_without_fpn.py │ │ ├── draw_box_utils.py │ │ ├── my_dataset.py │ │ ├── network_files/ │ │ │ ├── __init__.py │ │ │ ├── boxes.py │ │ │ ├── det_utils.py │ │ │ ├── faster_rcnn_framework.py │ │ │ ├── image_list.py │ │ │ ├── roi_head.py │ │ │ ├── rpn_function.py │ │ │ └── transform.py │ │ ├── pascal_voc_classes.json │ │ ├── plot_curve.py │ │ ├── predict.py │ │ ├── record_mAP.txt │ │ ├── requirements.txt │ │ ├── split_data.py │ │ ├── train_mobilenetv2.py │ │ ├── train_multi_GPU.py │ │ ├── train_res50_fpn.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── coco_eval.py │ │ │ ├── coco_utils.py │ │ │ ├── distributed_utils.py │ │ │ ├── group_by_aspect_ratio.py │ │ │ └── train_eval_utils.py │ │ ├── transforms.py │ │ └── validation.py │ ├── mask_rcnn/ │ │ ├── README.md │ │ ├── backbone/ │ │ │ ├── __init__.py │ │ │ ├── feature_pyramid_network.py │ │ │ └── resnet50_fpn_model.py │ │ ├── coco91_indices.json │ │ ├── det_results20220406-141544.txt │ │ ├── draw_box_utils.py │ │ ├── my_dataset_coco.py │ │ ├── my_dataset_voc.py │ │ ├── network_files/ │ │ │ ├── __init__.py │ │ │ ├── boxes.py │ │ │ ├── det_utils.py │ │ │ ├── faster_rcnn_framework.py │ │ │ ├── image_list.py │ │ │ ├── mask_rcnn.py │ │ │ ├── roi_head.py │ │ │ ├── rpn_function.py │ │ │ └── transform.py │ │ ├── pascal_voc_indices.json │ │ ├── plot_curve.py │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── seg_results20220406-141544.txt │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── coco_eval.py │ │ │ ├── coco_utils.py │ │ │ ├── distributed_utils.py │ │ │ ├── group_by_aspect_ratio.py │ │ │ └── train_eval_utils.py │ │ ├── transforms.py │ │ └── validation.py │ ├── retinaNet/ │ │ ├── README.md │ │ ├── backbone/ │ │ │ ├── __init__.py │ │ │ ├── feature_pyramid_network.py │ │ │ └── resnet50_fpn_model.py │ │ ├── draw_box_utils.py │ │ ├── my_dataset.py │ │ ├── network_files/ │ │ │ ├── __init__.py │ │ │ ├── anchor_utils.py │ │ │ ├── boxes.py │ │ │ ├── det_utils.py │ │ │ ├── image_list.py │ │ │ ├── losses.py │ │ │ ├── retinanet.py │ │ │ └── transform.py │ │ ├── pascal_voc_classes.json │ │ ├── plot_curve.py │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── results20210421-142632.txt │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── coco_eval.py │ │ │ ├── coco_utils.py │ │ │ ├── distributed_utils.py │ │ │ ├── group_by_aspect_ratio.py │ │ │ └── train_eval_utils.py │ │ ├── transforms.py │ │ └── validation.py │ ├── ssd/ │ │ ├── README.md │ │ ├── draw_box_utils.py │ │ ├── my_dataset.py │ │ ├── pascal_voc_classes.json │ │ ├── plot_curve.py │ │ ├── predict_test.py │ │ ├── record_mAP.txt │ │ ├── requirements.txt │ │ ├── src/ │ │ │ ├── __init__.py │ │ │ ├── res50_backbone.py │ │ │ ├── ssd_model.py │ │ │ └── utils.py │ │ ├── train_multi_GPU.py │ │ ├── train_ssd300.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── coco_eval.py │ │ │ ├── coco_utils.py │ │ │ ├── distributed_utils.py │ │ │ ├── group_by_aspect_ratio.py │ │ │ └── train_eval_utils.py │ │ ├── transforms.py │ │ └── validation.py │ ├── train_coco_dataset/ │ │ ├── README.md │ │ ├── backbone/ │ │ │ ├── __init__.py │ │ │ ├── feature_pyramid_network.py │ │ │ ├── mobilenetv2_model.py │ │ │ ├── resnet.py │ │ │ ├── resnet50_fpn_model.py │ │ │ └── vgg_model.py │ │ ├── change_backbone_with_fpn.py │ │ ├── coco91_indices.json │ │ ├── compute_receptive_field.py │ │ ├── draw_box_utils.py │ │ ├── my_dataset.py │ │ ├── network_files/ │ │ │ ├── __init__.py │ │ │ ├── boxes.py │ │ │ ├── det_utils.py │ │ │ ├── faster_rcnn_framework.py │ │ │ ├── image_list.py │ │ │ ├── roi_head.py │ │ │ ├── rpn_function.py │ │ │ └── transform.py │ │ ├── plot_curve.py │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── results20220408-201436.txt │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── coco_eval.py │ │ │ ├── distributed_utils.py │ │ │ ├── group_by_aspect_ratio.py │ │ │ └── train_eval_utils.py │ │ ├── transforms.py │ │ └── validation.py │ └── yolov3_spp/ │ ├── README.md │ ├── build_utils/ │ │ ├── __init__.py │ │ ├── datasets.py │ │ ├── img_utils.py │ │ ├── layers.py │ │ ├── parse_config.py │ │ ├── torch_utils.py │ │ └── utils.py │ ├── calculate_dataset.py │ ├── cfg/ │ │ ├── hyp.yaml │ │ └── yolov3-spp.cfg │ ├── draw_box_utils.py │ ├── export_onnx.py │ ├── load_onnx_test.py │ ├── models.py │ ├── predict_test.py │ ├── requirements.txt │ ├── results20210515-152935.txt │ ├── train.py │ ├── train_multi_GPU.py │ ├── train_utils/ │ │ ├── __init__.py │ │ ├── coco_eval.py │ │ ├── coco_utils.py │ │ ├── distributed_utils.py │ │ ├── group_by_aspect_ratio.py │ │ └── train_eval_utils.py │ ├── trans_voc2yolo.py │ └── validation.py ├── pytorch_segmentation/ │ ├── deeplab_v3/ │ │ ├── README.md │ │ ├── get_palette.py │ │ ├── my_dataset.py │ │ ├── palette.json │ │ ├── pascal_voc_classes.json │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── results20211027-104607.txt │ │ ├── src/ │ │ │ ├── __init__.py │ │ │ ├── deeplabv3_model.py │ │ │ ├── mobilenet_backbone.py │ │ │ └── resnet_backbone.py │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── distributed_utils.py │ │ │ └── train_and_eval.py │ │ ├── transforms.py │ │ └── validation.py │ ├── fcn/ │ │ ├── README.md │ │ ├── get_palette.py │ │ ├── my_dataset.py │ │ ├── palette.json │ │ ├── pascal_voc_classes.json │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── results20210918-122740.txt │ │ ├── src/ │ │ │ ├── __init__.py │ │ │ ├── backbone.py │ │ │ └── fcn_model.py │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── distributed_utils.py │ │ │ └── train_and_eval.py │ │ ├── transforms.py │ │ └── validation.py │ ├── lraspp/ │ │ ├── README.md │ │ ├── get_palette.py │ │ ├── my_dataset.py │ │ ├── palette.json │ │ ├── pascal_voc_classes.json │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── results20211028-105233.txt │ │ ├── src/ │ │ │ ├── __init__.py │ │ │ ├── lraspp_model.py │ │ │ └── mobilenet_backbone.py │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── distributed_utils.py │ │ │ └── train_and_eval.py │ │ ├── transforms.py │ │ └── validation.py │ ├── u2net/ │ │ ├── README.md │ │ ├── convert_weight.py │ │ ├── my_dataset.py │ │ ├── predict.py │ │ ├── requirements.txt │ │ ├── results20220723-123632.txt │ │ ├── src/ │ │ │ ├── __init__.py │ │ │ └── model.py │ │ ├── train.py │ │ ├── train_multi_GPU.py │ │ ├── train_utils/ │ │ │ ├── __init__.py │ │ │ ├── distributed_utils.py │ │ │ └── train_and_eval.py │ │ ├── transforms.py │ │ └── validation.py │ └── unet/ │ ├── README.md │ ├── compute_mean_std.py │ ├── my_dataset.py │ ├── predict.py │ ├── requirements.txt │ ├── results20220109-165837.txt │ ├── src/ │ │ ├── __init__.py │ │ ├── mobilenet_unet.py │ │ ├── unet.py │ │ └── vgg_unet.py │ ├── train.py │ ├── train_multi_GPU.py │ ├── train_utils/ │ │ ├── __init__.py │ │ ├── dice_coefficient_loss.py │ │ ├── distributed_utils.py │ │ └── train_and_eval.py │ └── transforms.py ├── summary_problem.md └── tensorflow_classification/ ├── ConfusionMatrix/ │ ├── class_indices.json │ ├── main.py │ └── model.py ├── ConvNeXt/ │ ├── model.py │ ├── predict.py │ ├── train.py │ ├── trans_weights.py │ └── utils.py ├── README.md ├── Test11_efficientnetV2/ │ ├── model.py │ ├── predict.py │ ├── train.py │ ├── trans_weights.py │ └── utils.py ├── Test1_official_demo/ │ ├── model.py │ └── train.py ├── Test2_alexnet/ │ ├── class_indices.json │ ├── fine_train_alexnet.py │ ├── model.py │ ├── predict.py │ ├── read_pth.py │ ├── train.py │ └── trainGPU.py ├── Test3_vgg/ │ ├── class_indices.json │ ├── fine_train_vgg16.py │ ├── model.py │ ├── predict.py │ ├── read_ckpt.py │ ├── train.py │ └── trainGPU.py ├── Test4_goolenet/ │ ├── class_indices.json │ ├── model.py │ ├── model_add_bn.py │ ├── predict.py │ ├── read_pth.py │ ├── train.py │ ├── trainGPU.py │ └── train_add_bn.py ├── Test5_resnet/ │ ├── batch_predict.py │ ├── class_indices.json │ ├── model.py │ ├── predict.py │ ├── read_ckpt.py │ ├── read_h5.py │ ├── subclassed_model.py │ ├── train.py │ └── trainGPU.py ├── Test6_mobilenet/ │ ├── model_v2.py │ ├── model_v3.py │ ├── predict.py │ ├── read_ckpt.py │ ├── trainGPU_mobilenet_v2.py │ ├── train_mobilenet_v2.py │ ├── train_mobilenet_v3.py │ ├── trans_v3_weights.py │ └── utils.py ├── Test7_shuffleNet/ │ ├── model.py │ ├── predict.py │ ├── train.py │ ├── trans_weights.py │ └── utils.py ├── Test9_efficientNet/ │ ├── model.py │ ├── predict.py │ ├── train.py │ └── utils.py ├── analyze_weights_featuremap/ │ ├── alexnet_model.py │ ├── analyze_feature_map.py │ └── analyze_kernel_weight.py ├── custom_dataset/ │ ├── train_fit.py │ └── utils.py ├── swin_transformer/ │ ├── model.py │ ├── predict.py │ ├── train.py │ ├── trans_weights.py │ └── utils.py ├── tensorboard_test/ │ ├── train_fit.py │ └── train_not_fit.py └── vision_transformer/ ├── predict.py ├── train.py ├── trans_weights.py ├── utils.py └── vit_model.py