jahongir7174 / YOLOv8-human

YOLOv8 re-implementation for human detection using PyTorch
GNU Affero General Public License v3.0
88 stars 13 forks source link

Difference between yolov8-pt and yolov8-human #6

Open ashish-roopan opened 1 year ago

ashish-roopan commented 1 year ago

@jahongir7174 Great job on the re-implementation, its really helpful for making changes to the model and training. I want to know what is the diffrence between your two repos " yolov8-pt" and yolov8-human"? Is there any architecture changes or is it just the dataset which is different?

jahongir7174 commented 1 year ago

@ashish-roopan you can find the NN changes in the detection head because of the number of classes. Other parts such as hyperparameters are the same as the original