YuTong Li (Harry)
- 12GB NVIDIA GPU
- Python 3.9.0
- PyTorch v1.12 or later
Note:
app.py
is the Streamlit launcher.
Before running, please make sure you have saved the trained model in the current directory.
MobileViTv3 is introduced in the paper:
MobileViTv3: Mobile-Friendly Vision Transformer with Simple and Effective Fusion of Local, Global and Input Features
by Shakti N. Wadekar and Abhishek Chaurasia
This project verifies the effectiveness of incorporating MobileViTv3 into YOLOv8:
- Results are better than the original
yolov8n.pt
- Training curves are smoother