logo

16 Commits (34415c2d642bb220cc42f9590ff173f7e7b0ca71)

Author SHA1 Message Date
ChengZi 266bcc0651 add converting to cpu 3 years ago
ChengZi 6f7eb6b56c remove get_model and pytorch folder. 3 years ago
zhang chen 086c225df0 fix embedding_extractor.py memory leak problem, and add a `num_classes` param in the construct function of operator. 3 years ago
zhang chen 51a17a4bb0 add custom optimizer and loss case 3 years ago
zhang chen 28cdc0f73a resnet op train from yaml config. 3 years ago
wxywb 951ae888e7 change the input format. 3 years ago
zhang chen 22aa3c7e26 cancel import warning 3 years ago
wxywb 14cacc0ee7 convert image to RGB image. 3 years ago
zhang chen b32cd33766 fix import problem 3 years ago
zhang chen 7929342d7a matching embedding test output with towhee op output 3 years ago
shiyu22 f6b8ce3ba7 Update 3 years ago
shiyu22 766b44c29e Update 3 years ago
shiyu22 6308ec7a76 Update 3 years ago
shiyu22 838738f354 Update 3 years ago
shiyu22 7aa9ecd53c Update 3 years ago
shiyu22 d00e84ef2c Update 3 years ago
shiyu22 c9d3dcd1fb Update 3 years ago
shiyu22 4adfae8f02 Update model 3 years ago
Kaiyuan Hu ce5e3339a5 Update and add gitignore 3 years ago
shiyu22 47f74b3813 Add resnet50-image-embedding op 3 years ago