enazoe / yolo-tensorrt

TensorRT8.Support Yolov5n,s,m,l,x .darknet -> tensorrt. Yolov4 Yolov3 use raw darknet *.weights and *.cfg fils. If the wrapper is useful to you,please Star it.
MIT License
1.18k stars 316 forks source link

Is it support Scaled-YOLOv4 ? #84

Open Scottchou7 opened 3 years ago

Scottchou7 commented 3 years ago

Scaled-YOLOv4 has already released. Is it supported?

Scaled-YOLOv4: Scaling Cross Stage Partial Network Scaled-YOLOv4 yolov4-csp.cfg

enazoe commented 3 years ago

is working on it

Scottchou7 commented 3 years ago

OK, Thanks!

DatDoc commented 3 years ago

@enazoe you're working on it ?

enazoe commented 3 years ago

@Scottchou7 @DatDoc yolov4-csp is ok now at branch yolov4_scaled

DatDoc commented 3 years ago

@enazoe Can it run on google colab ? Btw you're doing some great work, keep it up

auto-amc commented 3 years ago

@enazoe! Please update code to use last yolov4-csp.cfg from https://github.com/AlexeyAB/darknet

LukeAI commented 3 years ago

@enazoe is yolor in the works? https://github.com/WongKinYiu/yolor

auto-amc commented 3 years ago

Hi!i am getting the following errorUsing torch 1.9.0 CUDA:0 (GeForce GTX 1070, 8116MB) Traceback (most recent call last):  File "test.py", line 330, in     save_conf=opt.save_conf,  File "test.py", line 63, in test    model = Darknet(opt.cfg).to(device)  File "~/yolor/models/models.py", line 530, in init    self.module_defs = parse_model_cfg(cfg)  File "~/yolor/utils/parse_config.py", line 50, in parse_model_cfg    assert not any(u), "Unsupported fields %s in %s. See https://github.com/ultralytics/yolov3/issues/631" % (u, path)AssertionError: Unsupported fields ['objectness_smooth', 'resize', 'obj_normalizer', 'new_coords', 'max_delta'] in ~/darknet/cfg/yolov4-csp.cfg 20.06.2021, 20:43, "LukeAI" @.**@. is yolor in the works?https://github.com/WongKinYiu/yolor—You are receiving this because you commented.Reply to this email directly, view it on GitHub, or unsubscribe.