Closed striving1234 closed 6 months ago
updated gen_wts script to support yolov5 classfication model export
updated yolov5-s architecture to support classification head.
updated yolov5_trt infer script to support yolov5 classification model alongwith existing detection model.
added imagenet_classes file to load list of 1k classes required for the infer script.
final conv block doesn't require dynamic scale factor for out channel, hence hardcoded value provided to support all model varients
yolov5.cpp reverted back to original state, with explicit code changes
python infer script reverted back to prior state, with explicit code changes
seperate cpp file added for yolov5 classification task
cmake updated for yolo5 classification file
seperate python infer script for classification task
cmake updated by replacing cuda_add_executable with add_executable
default value added to the type argument
post-processing removed from yolov5 clasisifcation module
pre-processing for yolov5 classification inferencing
classification macro removed from the original yolov5 detection cpp file
reverted some extremely minor formatting changes.
reverted back to prior state by removing all formatting changes.
updated gen_wts script to support yolov5 classfication model export
updated yolov5-s architecture to support classification head.
updated yolov5_trt infer script to support yolov5 classification model alongwith existing detection model.
added imagenet_classes file to load list of 1k classes required for the infer script.
final conv block doesn't require dynamic scale factor for out channel, hence hardcoded value provided to support all model varients
yolov5.cpp reverted back to original state, with explicit code changes
python infer script reverted back to prior state, with explicit code changes
seperate cpp file added for yolov5 classification task
cmake updated for yolo5 classification file
seperate python infer script for classification task
cmake updated by replacing cuda_add_executable with add_executable
default value added to the type argument
post-processing removed from yolov5 clasisifcation module
pre-processing for yolov5 classification inferencing
classification macro removed from the original yolov5 detection cpp file
reverted some extremely minor formatting changes.
reverted back to prior state by removing all formatting changes.