How wzmiaomiao/deep-learning-for-image-processing works
This repository is an educational collection of image-processing and deep-learning examples, rather than one integrated application. Its principal workflow is to study a task or network, prepare data, build and train a model, then use prediction or evaluation examples; deployment and model-analysis projects extend that workflow. The README documents classification, detection, segmentation, instance segmentation, and keypoint detection. The graph groups those tutorial branches and selected internal components. Many implementations were not sampled, so their internal connections are omitted rather than inferred.
Main entry points: model.py MobileViT model · pytorch_object_detection Detection examples · models.py TextCNN model · det_utils.py Faster R-CNN utilities
Components 24
Classification
- Classification examples
pytorch_classification - MobileViT model
pytorch_classification/MobileViT/model.py - MobileViT config
pytorch_classification/MobileViT/model_config.py - MobileViT transformer
pytorch_classification/MobileViT/transformer.py - TensorFlow classification
tensorflow_classification - TextCNN model
others_project/textcnnKeras/models.py
Object Detection
- Detection examples
pytorch_object_detection - Faster R-CNN boxes
pytorch_object_detection/faster_rcnn/network_files/boxes.py - Faster R-CNN utilities
pytorch_object_detection/faster_rcnn/network_files/det_utils.py - Mask R-CNN boxes
pytorch_object_detection/mask_rcnn/network_files/boxes.py - Mask R-CNN utilities
pytorch_object_detection/mask_rcnn/network_files/det_utils.py - YOLO dataset loader
pytorch_object_detection/yolov3_spp/build_utils/datasets.py
Segmentation
- Segmentation examples
pytorch_segmentation
Keypoints
- Keypoint examples
pytorch_keypoint - DeepPose model
pytorch_keypoint/DeepPose/model.py - HRNet train evaluation
pytorch_keypoint/HRNet/train_utils/train_eval_utils.py - HRNet transforms
pytorch_keypoint/HRNet/transforms.py
Deployment
- ONNX export
deploying_service/deploying_pytorch/convert_onnx_cls/main.py - Flask inference service
deploying_service/deploying_pytorch/pytorch_flask_service/main.py - Model pruning
deploying_service/pruning_model_pytorch/main.py
Utilities
- Anchor clustering
others_project/kmeans_anchors/main.py - WIDERFace converter
others_project/trans_widerface_to_xml/main.py - Custom dataset example
pytorch_classification/custom_dataset/main.py
Other components
- Learner
Connections 18
- Learner → Classification examples: studies and trains
- Learner → TensorFlow classification: studies and trains
- Learner → Detection examples: studies and trains
- Learner → Segmentation examples: studies and trains
- Learner → Keypoint examples: studies and trains
- MobileViT model → MobileViT config: imports
pytorch_classification/MobileViT/model.py - MobileViT model → MobileViT transformer: imports
pytorch_classification/MobileViT/model.py - TextCNN model → Learner: supports text classification
- Faster R-CNN utilities → Faster R-CNN boxes: uses
pytorch_object_detection/faster_rcnn/network_files/rpn_function.py - YOLO dataset loader → Detection examples: provides training data
pytorch_object_detection/yolov3_spp/train_multi_GPU.py - Mask R-CNN utilities → Mask R-CNN boxes: uses
- HRNet train evaluation → HRNet transforms: imports
pytorch_keypoint/HRNet/train_utils/train_eval_utils.py - Learner → ONNX export: exports model
- Learner → Flask inference service: runs inference
- Learner → Model pruning: experiments with pruning
- Learner → Anchor clustering: tunes detector anchors
- Learner → WIDERFace converter: converts annotations
- Learner → Custom dataset example: learns data loading