Lahiru Rumesh Rathnayake's repositories
simple_cnn
Simple CNN is a library that can be used to train and infer CNN models by use of PyTorch and ONNX.
SSL_Vision_Pipeline
Semi-Supervised | Pseudo Labeling pipeline with one-stage object detection models
YOLOv3-weights-converter
convert trained yolov3 weights (.h5) model to TensorFlow severing
dataPipeline
data pipeline use for cnn
detectron2
Detectron2 is FAIR's next-generation platform for object detection and segmentation.
detr
End-to-End Object Detection with Transformers
kmeans-anchor-boxes
k-means clustering with the Intersection over Union (IoU) metric as described in the YOLO9000 paper
llm
An ecosystem of Rust libraries for working with large language models
ML_Data_Pre-Process
Data Pre-Processing for Machine Learning
pytorch-YOLOv4
PyTorch ,ONNX and TensorRT implementation of YOLOv4
semantic-segmentation
SOTA Semantic Segmentation Models in PyTorch
tensorflow-yolov4-tflite
YOLOv4, YOLOv4-tiny, YOLOv3, YOLOv3-tiny Implemented in Tensorflow 2.0, Android. Convert YOLO v4 .weights tensorflow, tensorrt and tflite
vision-transformer
Tensorflow implementation of the Vision Transformer (An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale)
yolov3-tf2
YoloV3 Implemented in Tensorflow 2.0