How to Train a Custom Faster R-CNN Model with Facebook AI's Detectron2 | Use Your Own Dataset

แชร์
ฝัง
  • เผยแพร่เมื่อ 5 ก.พ. 2025

ความคิดเห็น • 41

  • @Baddogbling321
    @Baddogbling321 ปีที่แล้ว +8

    HI, the colab notebooks are very outdated to the point nothing works. PLEASE UPDATE THE COLAB NOTEBOOKS ROBOFLOW! Thanks.

  • @Mee-yp5in
    @Mee-yp5in 2 ปีที่แล้ว +1

    A very fine platform for preparing dataset and training model, recommended!!

  • @athlete645
    @athlete645 4 ปีที่แล้ว +3

    When the model was trained, shouldn't have the cfg.MODEL.ROID_HEADS.NUM_CLASSES been changed? The comment on that line says "your number of classes +1".
    Thank you for the great explanation! This was exactly the video I needed.

  • @GaJolGuru
    @GaJolGuru ปีที่แล้ว +1

    Hi Roboflow
    Thanks for this great video, can you perahps help me out here. I used the workflow and updated both torch and torchvision as these old packages are not available anymore. However detectron2 seems not to be found anywhere, any ideas?
    Thanks

  • @shoebbashar4385
    @shoebbashar4385 ปีที่แล้ว +2

    Please update the colab notebook to tensorflow latest version.

  • @cheerio4742
    @cheerio4742 ปีที่แล้ว +1

    error when installing torch and torchvision in their version "Could not find a version that satisfies the requirement torch==1.5"

  • @tutsy98
    @tutsy98 2 ปีที่แล้ว

    THANK U ROBOFLOWWWWW

  • @omerkaya5669
    @omerkaya5669 ปีที่แล้ว +1

    I have several questions?
    How can I plot PR curve and ROC curve?
    I also want to reach the image of confusion matrix.
    Where can I find the F score value and mAP values?

    • @Roboflow
      @Roboflow  ปีที่แล้ว

      Unfortunately, those metrics don't come out of the box with Detectron2. However, you can calculate them using other packages, for example: github.com/SkalskiP/onemetric

    • @omerkaya5669
      @omerkaya5669 ปีที่แล้ว

      @@Roboflow Thank you for answer. Roboflow is the best learning page

    • @omerkaya5669
      @omerkaya5669 ปีที่แล้ว

      @@Roboflow confusion_matrix = ConfusionMatrix.from_detections(
      true_batches=true_batches,
      detection_batches=detection_batches,
      num_classes=1
      )
      I run this code. but i take this error.
      NameError: name 'ConfusionMatrix' is not defined

  • @minsoojeong199
    @minsoojeong199 4 ปีที่แล้ว +1

    Hello, thank you for the good lecture. I'd like to know how many photos I can upload to roboflow.

  • @_ABDULGHANI
    @_ABDULGHANI 3 ปีที่แล้ว

    Hello there and thank you, if I want to train for longe time is this method support continue from the last checkpoint.

  • @himomhey3437
    @himomhey3437 2 ปีที่แล้ว

    Good Tutorial!
    How can I deploy the model.pth in a raspberry pi 4

  • @akshayakki5692
    @akshayakki5692 ปีที่แล้ว

    Sir, is it necessary to annotate test images also or why we should have annotated labels for test images those images are for only testing right.

  • @danaali1710
    @danaali1710 3 ปีที่แล้ว

    Hi, thank you please answer my question. if my project consists of detecting one class of object , should i annotate my images and create bounding box.

  • @Fatima-kj9ws
    @Fatima-kj9ws 3 ปีที่แล้ว

    Great, very well explained tutorial. Thank you very much

    • @Ballibhai007
      @Ballibhai007 2 ปีที่แล้ว

      i have question are you free ???

  • @alhasanalshaebi
    @alhasanalshaebi 4 ปีที่แล้ว +1

    How does one choose an optimal number of epochs?especially in this case of Dataset.

  • @tanzimmostafa
    @tanzimmostafa 2 ปีที่แล้ว

    Hello, can someone please tell me how to find out the mAP metric. The notebook only shows the AP for each object class.

  • @__-dr2if
    @__-dr2if 10 หลายเดือนก่อน

    the code uses tensorflow_gpu 1.5 but now google colab supports 2.8.0 or above and so not able t download API

  • @drswapnil1234
    @drswapnil1234 2 ปีที่แล้ว

    Hi. I need to detect circles in a custom model. I have annonated the training images using labelme. It gives 2 no pairs in the json. Can you pls help in how to modify the code for circle detection. Its giving an error as not polygon with current code. Thanks

  • @altamontholmes8719
    @altamontholmes8719 3 ปีที่แล้ว

    Hi, how can I visualize the feature maps? Btw thanks for the good and brief tutorial!

  • @pavan3078
    @pavan3078 ปีที่แล้ว

    anyone knows what is happening here
    ERROR: Could not find a version that satisfies the requirement torch==1.5 (from versions: 1.11.0, 1.12.0, 1.12.1, 1.13.0, 1.13.1, 2.0.0, 2.0.1, 2.1.0)
    ERROR: No matching distribution found for torch==1.5

    • @afwanya
      @afwanya 8 หลายเดือนก่อน

      Yeah same. Did you already fix it?

  • @Paperwhite786
    @Paperwhite786 4 ปีที่แล้ว +7

    This is a Roboflow ad

  • @anirudhputsala8276
    @anirudhputsala8276 3 ปีที่แล้ว

    Hi, After generating the dataset, the train folder neither show any preview of the images nor has a folder of it's name in the download. Is there any limit of images I could train without pro version?

  • @melvin1919
    @melvin1919 3 ปีที่แล้ว

    When i run , the class names are not showed up? any reason why?

  • @rohanrane7835
    @rohanrane7835 3 ปีที่แล้ว

    Hi, it is showing me import error when executing "from detectron2 import model_zoo".
    How do I fix this?

  • @zubairsk1624
    @zubairsk1624 2 ปีที่แล้ว

    can any one please tell me how to train it on my local GPU instead of colab notebook

  • @vigneshbalaji6826
    @vigneshbalaji6826 4 ปีที่แล้ว

    How to use multi-gpu in detectron2 for custom data training

  • @JOHN-vb5bh
    @JOHN-vb5bh 3 ปีที่แล้ว

    Can anyone give suggestions or instructions on integrating this code with detectron2 official code. I had done
    from detectron2.engine import MyTrainer
    from detectron2.engine import LossEvalHook
    and
    trainer = MyTrainer(cfg)
    in my training scripts
    But getting an error
    TypeError Traceback (most recent call last)
    in
    21
    22 os.makedirs(cfg.OUTPUT_DIR, exist_ok=True)
    ---> 23 trainer = MyTrainer(cfg)
    24 trainer.resume_or_load(resume=False)
    25 trainer.train()
    TypeError: 'module' object is not callable
    Please help me to solve this problem

  • @neelkanthbhavnagarwala6001
    @neelkanthbhavnagarwala6001 2 ปีที่แล้ว

    SizeMismatchError when trying to train a model on custom dataset

  • @darkamax1
    @darkamax1 3 ปีที่แล้ว

    Hello,
    Thank you for the amazingly well explained tutorial. I wrote a code and am using detectron2 to identify objects. It is working great with my in built webcam but whenever I try and use my webcam, it displays only the first frame and then says ..."ASSERT: "false" in file qasciikey.cpp, line 501"
    My Hardware config is as follows
    Ubuntu 18.04
    Nvidia GeForce GTX 1650Ti Card
    Torch Version '1.8.1+cu102'
    Cuda 10.2
    Python 3.6.9
    Any help will be much appreciated.
    Thanks in advance.

    • @darkamax1
      @darkamax1 3 ปีที่แล้ว

      Also my code is unable to read any key inputs when it is running, when attempting to quit the code with
      if ord=='q':
      cv2.destroyAllWindows()
      it either hangs or dumps the code with error
      Segmentation fault, core dumped
      Thanks

  • @hailuu625
    @hailuu625 3 ปีที่แล้ว

    Is there anyone got error KeyError: 'instances'

    • @janeng91
      @janeng91 3 ปีที่แล้ว

      Did you trained this model? How long for the training time?

    • @neelkanthbhavnagarwala6001
      @neelkanthbhavnagarwala6001 2 ปีที่แล้ว

      sizeMismatchError when trying to train a model using custom dataset..

  • @DiepHuongTran
    @DiepHuongTran 2 ปีที่แล้ว

    hello sir. cfg.MODEL.ROI_HEADS.NUM_CLASSES = 4 # your number of classes + 1. If I train 6 classes, set it to 7 right?

  • @vigneshbalaji6826
    @vigneshbalaji6826 4 ปีที่แล้ว

    How to use multi-gpu in detectron2 for custom data training