From d0d4a40b0bebdf2637933fd2b5ce7887c674a6c7 Mon Sep 17 00:00:00 2001 From: Deepankar Sharma <74599435+ideepankarsharma2003@users.noreply.github.com> Date: Fri, 3 Jan 2025 16:07:47 +0530 Subject: [PATCH] Update README.md --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index 518cd7a..deaafad 100644 --- a/README.md +++ b/README.md @@ -50,6 +50,8 @@ python convert_y3.py model_data\weights\yolov3.weights model_data\models\yolov3. python convert_y4.py model_data\weights\yolov4.weights model_data\models\yolov4.h5 ``` +- Download [RESNET-50](https://download.pytorch.org/models/resnet50-19c8e357.pth) and save it as `model_data/models/model.pth`. + # Pre-trained models (.h5) (If you want to start right away) - Download the Keras models for [YOLO_v3](https://drive.google.com/file/d/1a7JI-A920lrdt6OKya-qCXx-5ZUWvkMg/view?usp=sharing) and [YOLO_v4](https://drive.google.com/file/d/1pwFo4aHKPi0ztpL5tEYaXIr8RltYYQeY/view?usp=sharing) and add them to \model_data\models\