Skip to content

Building the model #29

Open
Open

Description

@Tomysinee

Expected behavior

I think the model should start training

Actual behavior

Error when trying to build the model with/without the pre-trained weights (following the building instruction)

Steps to reproduce behavior

with yolo weights

without weights1

Activity

  1. anivegesana commented on Mar 31, 2021

    @anivegesana
    Collaborator

    Hi @Tomysinee,

    Thank you for your interest in our project! We are actually still in the development phase of the YOLO model. We have yet to match the accuracy of the paper when trained from scratch.

    It is hard to tell what the issue is without looking at your build.py, but I am assuming that you are trying to train YOLOv3 from the pretrained weights given in the original paper.

    We have a configuration for training YOLOv3, but I do not remember where the original weights came from. We will be able to give you more detailed information when we completely reproduce the paper and clean up the repository, but you can definitely use the weights from the author's website by changing csdarknet53 in the config to yolov3.

    python3 -m yolo.train_vm --mode=train_and_eval --experiment=yolo_custom --model_dir=checkpoints/yolov3 --config_file=yolo/configs/experiments/yolov3/yolov3.yaml

    If you are trying to train YOLOv4 or something else, feel free to reply.

  2. Tomysinee commented on Mar 31, 2021

    @Tomysinee
    Author

    Thank you so much for your reply.

    The build.py is the building instruction in the repo. I have attached an image with respect to that. . Also, I am trying to train (using Yolov4) from scratch on an input image with a different number of channels than the original paper
    build

  3. anivegesana commented on Apr 1, 2021

    @anivegesana
    Collaborator

    Hi @Tomysinee,

    I would still recommend holding off on training with our model implementation since we are not done.

    If you do want to train the model from scratch using our repo, you can make modifications to the yolov4.yaml experiment that we provide and use the command that I gave earlier to train. The weights for the CSPDarknet model that we start from can be downloaded here. You need to adjust the number of training steps and validation steps for your experiment. For most datasets that have a similar distribution as COCO, you can scale them by the number of images in the dataset. An example for how to use the model for inference is in the inference.ipynb file.

    For now, if you want to have a model that matches the paper's accuracy, I would suggest training using darknet until we finish the YOLO model.

  4. kalikhademi commented on Jun 11, 2021

    @kalikhademi

    do we need to run the build.py to build the model before training or we can use your .yaml files to start the training?
    When I run the build.py, I got the same error as mentioned above.

  5. Tomysinee commented on Jun 11, 2021

    @Tomysinee
    Author
  6. kalikhademi commented on Jun 11, 2021

    @kalikhademi

    do we need to make changes in the .yaml file. I have used the train command mentioned above and it resulted in this error.
    Screen Shot 2021-06-11 at 11 44 14 AM

  7. Tomysinee commented on Jun 11, 2021

    @Tomysinee
    Author
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions